keras-retinanet 用自己的資料集訓練

Blackrosetian發表於2019-03-19

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

相關文章