注:第一次写博客,记录一下
1.下载作者大神的代码:
pytorch
2.下载EfficicentNet的预训练模型,最好是使用,我下载的是efficientdet-d2.pth
3.数据集放的位置
datasets
coco
annotations
train2017
val2017
4.在projects/coco.yml里面替换自己训练的名字:
mean和std替换自己的
5.在efficientdet/config.py 把类别进行修改:COCO_CLASSES
6.开始训练
python train.py -c 2 --batch_size 12 --lr 1e-4 --num_epochs 20 --load_weights logs/coco/efficientdet-d2.pth
![在这里插入图片描述](https://img-blog.csdnimg.cn/20200415123553630.png?x-oss-process=image/watermark,type_ZmxXzQyNjgwMzIx,size_16,color_F
7.进行评估:
python coco_eval.py -p coco -c 2 -w /logs/coco/efficientdet-d2_6_35973.pth
8.进行测试
python efficientdet_test.py
注:训练、评估和测试需要修改一些自己的名称
注:如有侵权还忘告知