修改了源码(之前可以正常运行),之前程序读取的标签集合路径是
F:\centernet-pytorch-main\VOCdevkit/VOC2007/JPEGImages/00017462fd484921ead12818c60c53b7.jpg 11,115,132,183,0 131,111,693,182,0 ……
这样的格式 其中0表示类别,前面四位是左上和右下的坐标
现在要改成用上四个点的坐标,我重新生成了标签集合文件:
F:\centernet-pytorch-main\VOCdevkit/VOC2007/JPEGImages/00017462fd484921ead12818c60c53b7.jpg 15,116,132,115,131,182,11,183,0 132,115,690,111,693,175,131,182,0 从左上右下两点变成了四个点的坐标(为什么改成四个点,因为模型改动里需要用上四个点)
然后把文件里原来读取标签和dataloader的地方都修改了 但感觉还是有问题 :
开始训练的时候在start train卡了好久 第一个epoch都没加载进去 然后报错;Caught MemoryError in DataLoader worker process 0.
看了写网上的教程我把num_workers改成了0 然后就一直卡在这里,直到显示内存不够闪退了
感觉是dataloader里加载数据有问题,求高人指点