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 ·
1479 阅读 ·
0 评论