前提
使用代码https://github.com/fizyr/keras-retinanet
参考博客https://www.freesion.com/article/6806110223/
环境ubuntu,tensorflow_gpu 1.14.0,Keras 2.3.1
数据集什么的都准备好了,进入./bin目录下运行训练命令。
训练
python train.py --batch-size 4 --steps 104 --epochs 30 csv ...<
在Ubuntu环境下,利用tensorflow_gpu 1.14.0和Keras 2.3.1,通过github上的keras-retinanet库训练RetinaNet模型。训练时遇到XML数据中坐标类型不一致的问题,将int改为float后成功训练。模型转换及测试阶段,调整了score阈值以改善检测效果,最终在修改test.py中得分阈值后,检测性能显著提升。
前提
使用代码https://github.com/fizyr/keras-retinanet
参考博客https://www.freesion.com/article/6806110223/
环境ubuntu,tensorflow_gpu 1.14.0,Keras 2.3.1
数据集什么的都准备好了,进入./bin目录下运行训练命令。
python train.py --batch-size 4 --steps 104 --epochs 30 csv ...<

被折叠的 条评论
为什么被折叠?