Retinanet训练自己的数据集
Retinanet训练Retinanet训练之前配置的retinanet环境忽然调用不了GPU,尝试多数方法未解决,然后重新配置。一.环境配置使用的环境是ubuntu18.04,CUDA10.1。下载Retinanet项目git clone https://github.com/fizyr/keras-retinanet.git编译python setup.py build_ext --inplace环境配置pip install -r requirements.txt
原创
2021-07-06 17:03:59 ·
1360 阅读 ·
0 评论