i4yyds
i4yyds
您好，我尝试将输入尺寸从600修改为800，然后计算loss的代码报错 loss[pos_mask] = -math.inf # 无穷 IndexError: The shape of the mask [1, 67995] at index 1 does not match the shape of the indexed tensor [1, 120087] at index...
> @SoulCoderCN 目前看来应该是需要手动调整的，只是我还没有找到相关联的地方在哪里 好像不需要手动调整，我刚刚用这个代码成功跑了VOC2007，而VOC2007中的图片尺寸并不是统一的。 目前训练自己的数据集还是不行，不知道出了什么问题。
> @SoulCoderCN 请问下你的显存是多少呀，我之前运行的时候，600尺度的图片 一直是 cuda out of memory 在config.py文件中设置 _C.TRAIN.BATCH_SIZE = 1，就不会溢出了
> @SoulCoderCN 输入大小改变，会导致特征图大小发生改变，你需要对 fpn的输入特征图大小进行更改，也就是_C.MODEL.ANCHORS.FEATURE_MAPS 项， 其值为 第一项为 IMAGE_SIZE/ （2**3）向上取整，其后每项均/2 向上取整；你的报错 IndexError: The shape of the mask [1, 67995] at index 1 does not match the shape of the indexed tensor...
torch=0.4.1 torchvision=0.2.1 After using the pre-training VGG model you provided, the error was reported as follows: Traceback (most recent call last): File "F:/DSS-pytorch-master2/main.py", line 95, in main(config) File "F:/DSS-pytorch-master2/main.py", line...
confidence loss ： 1) A = 有物体的格子 * (预测 - IOU) #目的是让A减小，即预测=IOU 2) B = 没有物体的格子 * 预测 #目的是让B减小，即没有物体的格子预测值为0 class loss： C = 有物体的格子 * (预测 - 真实值) #目的是让C减小，即预测=真实值 box...
> 为什么 读数据 没用到xml文件 用到了 /utils/pascal_voc.py 中的load_pascal_annotation函数
> > 去他娘的英文！ > > 我对代码中的loss层有4个疑问，目前已经解决3个。分享一些自己的观点，一起交流！ > > > > 1. 为什么使用 offset ？ > > 2. 为什么对 w和h 开根号 ？ > > 3. 为什么使用 offset_tran ？ > > 4....
> I get it. The network predicts the coordinate of box on one pixel. After reshape 49 pixels to 7x7, we need to get the box's coordinate on the 7x7...