使用命令 “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文件或重新下載數據集。