Pytorch 目标检测算法(1)(基于李沐老师的课程)

发布时间:2024-01-22 17:30

最近从0开始学习了李沐老师的动手学深度学习课程,收益匪浅,想着弄个笔记,若有错误请及时指出,感谢各位!

目录

最近从0开始学习了李沐老师的动手学深度学习课程,收益匪浅,想着弄个笔记,若有错误请及时指出,感谢各位!

1.预备基础

1.1 坐标变换

1)从(左上, 右下) ---> (中心位置, 宽度, 高度)

2) 从(中心位置, 宽度, 高度) ---> (左上, 右下)

1.2 矩形框(锚框)绘制


1.预备基础

1.1 坐标变换

对于图片而言,一般采取两种格式来保存图片中的位置信息(左上, 右下)(中心坐标, 宽, 高)

1)从(左上, 右下) ---> (中心位置, 宽度, 高度)

假设box = [Xl, Yl, Xr, Yr](其中l=left, r=right),那么

center = ((Xl + Xr) / 2, (Yl + Yr) / 2  

width = Xr - Xl

height=Yr-Yl。

这里的细节在于 Y轴的方向和我们平时生活中的不一样。他是从上到下递增,因此Yr为较大值。图示:

 \"Pytorch

具体实现:

def box_corner_to_center(boxes):
    \"\"\"从(左上, 右下)转换到(中间, 宽度, 高度)\"\"\"
    x1, y1, x2, y2 = boxes[:, 0], boxes[:, 1], boxes[:, 2], boxes[:, 3]
    cx = (x1 + x2) / 2  # 得到x的中心位置
    cy = (y1 + y2) / 2  # 得到y的中心位置
    w = x2 - x1  # 宽度
    h = y2 - y1  # 高度 
    boxes = torch.stack((cx, cy, w, h), axis=-1)
    return boxes

2) 从(中心位置, 宽度, 高度) ---> (左上, 右下)

假设box = [centerX, centerY, width, height], 那么

Xl = centerX - (0.5 * width)

Yl = centerY - (0.5 * height)

Xr = centerX + (0.5 * width)

Yr = centerY + (0.5 * height)

具体实现:

def box_center_to_corner(boxes):
    \"\"\"从(中间, 宽度, 高度)转换到(左上, 右下)\"\"\"
    cx, cy, w, h = boxes[:, 0], boxes[:, 1], boxes[:, 2], boxes[:, 3]
    xl = cx - 0.5 * w
    yl = cy - 0.5 * h
    xr = cx + 0.5 * w
    yr = cy + 0.5 * h
    boxes = torch.stack((xl, yl, xr, yr), axis=-1)
    return boxes

1.2 矩形框(锚框)绘制

可以运用opencv中的cv2.rectangle 或者 matplotlib.pylot中的Rectangle进行绘制,由于课程中使用的是matplotlib,为了后续内容这里只解释后者。

plt.Rectangle((Xl, Yl), width, height, fill, edgecolor, linewidth)

第一个参数:左上点

第二,三个参数:宽度,高度

fill: 是否填充矩形

edgecolor : 边框颜色

linewidth : 边框线的宽度 

具体实现:

def bbox_to_rect(bbox, color):
    return plt.Rectangle(xy=(bbox[0], bbox[1]),
                         width=bbox[2] - bbox[0],
                         height=bbox[3] - bbox[1],
                         fill=False,
                         edgecolor=color, linewidth=2)
 
 

ItVuer - 免责声明 - 关于我们 - 联系我们

本网站信息来源于互联网,如有侵权请联系:561261067@qq.com

桂ICP备16001015号