Tensorflow版本需要1.14.0及以上。之前用的是1.13.1,后面重装了1.14.0
1. github:https://github.com/fizyr/keras-retinanet
git clone https://github.com/fizyr/keras-retinanet.git
2. follow README
--> 编译环境 ①pip install numpy --user ② pip install . --user(这里需要稍微等一下) ③python setup.py build_ext --inplace(这里也要等一下)
3. 准备自己的数据,我这里用的是pascal VOC数据,数据组织如下:
VOC2007
--JPEGImages(训练数据,图片.jpg)
--Annotations(标签数据,.xml)
--ImageSets
--Main
--test.txt(测试的图片名,没有后缀)
--trainval.txt(训练的图片名,没有后缀)
4. 修改参数并训练、验证
①需要修改pascal_voc.py 里面的voc_classes类别,这里不包括背景。
②根据自己的数据修改epoch,steps per epoch,(steps≈训练样本数,epoch=总迭代次数/steps)
③根据检测目标大小修改anchor:在anchors.py下。
注:一个epoch之后会做一次evaluation,给了AP和F1精度评定指标