Yolov1源码 笔记记录

1.coord_mask = target_tensor[..., 4] > 0

Tensor 省略号(三个点)切片_tensor切片中...是什么意思_swust_fang的博客-CSDN博客

三个点自动判断维度

2.noobj_mask = noobj_mask.bool() #不是已经bool了?

3.bbox_pred = coord_pred[:, :5*B].contiguous().view(-1, 5) #防止内存不连续报错

4.image = cv2.imread(image_path)#某些老的图像处理软件使用的是 BGR 格式,因此 OpenCV 采用 BGR 格式可以与这些软件兼容。

img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB) # assuming the model is trained with RGB images.

5.

老笔记Yolo v1 笔记_Rondox的博客-CSDN博客

源码 讲解链接

参考

yolov1-代码精解_哔哩哔哩_bilibili

自己理解

Yolov1 源码讲解 voc.py_Rondox的博客-CSDN博客

Yolov1 源码讲解 loss.py_Rondox的博客-CSDN博客

Yolov1 源码讲解 train.py_Rondox的博客-CSDN博客

Yolov1 源码讲解 detect.py_Rondox的博客-CSDN博客

net部分跳过 暂时不管了

猜你喜欢

转载自blog.csdn.net/qq_36632604/article/details/130471829