github:https://github.com/fizyr/keras-retinanet
首先制作自己的数据集:将自己的数据集转换成VOC2007格式,再把VOC2007转化成CSV格式。
首先将自己的数据集转换成VOC2007格式的数据集:https://blog.csdn.net/u012426298/article/details/80334292
再将VOC2007格式的数据集转化为CSV:https://blog.csdn.net/u012426298/article/details/80343500
训练:python3 keras_retinanet/bin/train.py --weights /snapshots/resnet50_coco_best_v2.1.0.h5 --gpu 4,5,6,7 csv /images/annotations.csv /images/classes.csv
python3 keras_retinanet/bin/train.py csv /path/***/annotations.csv /path/***/classes.csv