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 源码讲解 voc.py_Rondox的博客-CSDN博客
Yolov1 源码讲解 loss.py_Rondox的博客-CSDN博客
Yolov1 源码讲解 train.py_Rondox的博客-CSDN博客
Yolov1 源码讲解 detect.py_Rondox的博客-CSDN博客
net部分跳过 暂时不管了