深度学习(七十二)ssd物体检测

深藏阁楼爱情的钟 2022-05-30 05:18 195阅读 0赞

def ssd_anchor_one_layer(img_shape,
                             feat_shape,
                             sizes,
                             ratios,
                             step,
                             offset=0.5,
                             dtype=np.float32):
        # 计算每个default center归一化后的坐标，因为是center，所以要加offset
        y, x = np.mgrid[0:feat_shape[0], 0:feat_shape[1]]
        y = (y.astype(dtype) + offset) * step / img_shape[0]
        x = (x.astype(dtype) + offset) * step / img_shape[1]
    
    
        y = np.expand_dims(y, axis=-1)
        x = np.expand_dims(x, axis=-1)
    
        num_anchors = len(sizes) + len(ratios)
        h = np.zeros((num_anchors, ), dtype=dtype)
        w = np.zeros((num_anchors, ), dtype=dtype)
    
    
        #第一种默认的default box的情况，也就是正方形，正方形的边长为sk
        h[0] = sizes[0] / img_shape[0]
        w[0] = sizes[0] / img_shape[1]
        di = 1
    
        #第二种默认的default box情况，也是正方形，只不过正方形的边长为sqrt(sk * sk+1)
        if len(sizes) > 1:
            h[1] = math.sqrt(sizes[0] * sizes[1]) / img_shape[0]
            w[1] = math.sqrt(sizes[0] * sizes[1]) / img_shape[1]
            di += 1
    
        #第三种~第六种default box，矩形边长比例为2、0.5,3,1/3的情况，对于conv4、和最后两层没有3、1/3两种default box
        for i, r in enumerate(ratios):
            h[i+di] = sizes[0] / img_shape[0] / math.sqrt(r)
            w[i+di] = sizes[0] / img_shape[1] * math.sqrt(r)
        return y, x, h, w

发表评论取消回复

表情：

评论列表（有 0 条评论，195人围观）

还没有评论，来说两句吧...

相关阅读

相关目标检测之：SSD

SSD300：网络结构图 ![watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmN

亦凉/ 2023年02月14日 02:29/ 0 赞/ 3 阅读

相关 TensorFlow2深度学习实战（十）：目标检测网络 SSD 论文解析

> 写在前面：大家好！我是【AI 菌】。我`热爱AI、热爱分享、热爱开源`！这博客是我对学习的一点总结与思考。如果您也对 `深度学习、机器视觉、算法、Python、C++`

- 日理万妓/ 2022年12月22日 00:46/ 0 赞/ 232 阅读

相关物体检测系列

物体检测系列一、物体检测网络有哪些 Anchor based： RCNN Fast RCNN Faster RCNN SPPNetS SDD

男娘i/ 2022年12月19日 09:20/ 0 赞/ 215 阅读

相关物体检测--HOG特征

有些长，转过来有时间看一、理论 1、HOG特征描述子的定义： locally normalised histogram of gradient orient

喜欢ヅ旅行/ 2022年09月30日 11:49/ 0 赞/ 568 阅读

相关深度学习(七十二)ssd物体检测

def ssd_anchor_one_layer(img_shape, feat_shap

深藏阁楼爱情的钟/ 2022年05月30日 05:18/ 0 赞/ 196 阅读

相关深度学习(七十三)pytorch学习笔记

先吐槽一下，深度学习发展速度真的很快，深度学习框架也在逐步迭代，真的是苦了俺这搞深度学习程序员。本人从三年前开始学深度学习开始，这些深度学习框架也是一个换过一个，从keras、

女爷i/ 2022年05月29日 13:54/ 0 赞/ 353 阅读

相关深度学习(七十二)tensorflow 集群训练

encoding:utf-8 -- coding: utf-8 -- 使用说明：1、修改分类数目;2、修改输入图片大小； 3、修改是否启用集

Love The Way You Lie/ 2022年05月29日 10:36/ 0 赞/ 207 阅读

相关如何快速选择最合适的物体检测框架：一个基于深度学习物体检测算法的简单测评

[如何快速选择最合适的物体检测框架：一个基于深度学习物体检测算法的简单测评][Link 1] 本文的主要内容来自于Google的一篇2017年的CVPR论文，“Speed/a

今天药忘吃喽~/ 2022年04月10日 07:10/ 0 赞/ 256 阅读

相关深度学习--简述卷积层如何检测物体边缘原理

对计算机视觉与深度学习有所了解的同学应该知道图像卷积可以识别出物体，在神经网络的前几层，识别出物体最基础的例如垂直或是水平的边缘。在之后的隐藏层慢慢可以识别出物体的部分，直到最

忘是亡心i/ 2022年03月11日 02:38/ 0 赞/ 247 阅读

相关【SSD目标检测】1：图片、视频内的物体检测与定位

一、SSD用于图片物体的定位与检测 [SSD原理介绍][SSD]这一篇博客对我的帮助比较大，很详细的介绍了SSD原理，送给大家做了解 1、下载SSD框架源码 1.

曾经终败给现在/ 2021年08月19日 15:58/ 1 赞/ 609 阅读