bug报错不一样

Question

老师我用的自己做的数据集的时候，如果samples_per_gpu=4,
workers_per_gpu=2,就会报DataLoader worker (pid 12432) is killed by signal: Terminated.
而 samples_per_gpu=2,
workers_per_gpu=2,报这个错ValueError: need at least one array to concatenate
然后取中值3的时候，报RuntimeError: DataLoader worker (pid 12947) is killed by signal: Terminated.
出现这种情况要怎么调试

Answer 1

会写代码的好厨师 2021-12-14 15:23:19

有救，数据做拼接的时候应该是用了最后一个纬度，3通道数据变成单通道以后，最后一个纬度默认删掉了。所以，方法是把最后一个纬度补上，也就是数据从 w*h变成w*h*1保留最后一个纬度

• 提问者 2hy #1

  感谢老师回复，我尝试打印了一下图片形状，好像没问题，数据集1.shape: (540, 960, 3)
  来自voc数据集000005.shape: (375, 500, 3)
  数据集2.shape (720, 1280, 3)
  是不是我代码VOC0712.py改的有问题，自制的数据集我给复制了两份，分别覆盖了2007和2012的文件夹，然后txt文件都指定到了一个文件

• 提问者 2hy #2

  非常感谢老师！

Answer 2

提问者 2hy 2021-12-08 09:25:54

老师，当时做数据集时不懂，把3通道处理成1了保存标注，还有的救么

• 提问者 2hy #1

  其它都检查过了，没问题，就是通道不一样，一换数据集就报错

• 提问者 2hy #2

  老师，数据集重新做了一个，还是不行，还是同样的报错

• 提问者 2hy #3

  单通道的问题可以解决，单通道的问题可以用cv转，但是还是报错就很难搞