CenterNet 在windows上训练出现错误

使用命令 “python main.py ctdet --exp_id coco_dla --batch_size 8 --lr 1.25e-4  --gpus 0 ” 训练，出现

错误1：

Traceback (most recent call last):
  File "<string>", line 1, in <module>
  File "C:\Anaconda3\envs\CenterNet\lib\multiprocessing\spawn.py", line 105, in spawn_main
    exitcode = _main(fd)
  File "C:\Anaconda3\envs\CenterNet\lib\multiprocessing\spawn.py", line 115, in _main
    self = reduction.pickle.load(from_parent)
EOFError: Ran out of input

解决方法：

在命令后加 --num_workers 0

参考“https://github.com/xingyizhou/CenterNet/issues/126”

 

错误2：

AttributeError: 'NoneType' object has no attribute 'shape'

解决方法：

训练图像不完整，修改训练的json文件或重新下载数据集。

参考“https://github.com/xingyizhou/CenterNet/issues/232”