项目地址:https://github.com/amdegroot/ssd.pytorch
这个是大佬重写的,方便阅读学习,所以作一遍,准备看代码
很容易装,安装python-opencv后运行报错。"cvImageShow"报错
坑:按照提示安装库,这个是不解决问题的
正确方法:重新安装opencv
pip install opencv-python -i https://pypi.douban.com/simple
问题解决
报错:
打开train.py,定位到93行,将loss.data[0]替换为loss.item(),再运行,成功
问题2:
CUDA out of memory
config.py中,修改batch_size=16