参考:https://blog.csdn.net/weixin_44791964/article/details/106343014
源码下载
https://github.com/bubbliiiing/efficientdet-pytorch
把自己的数据集放到如下位置
在根目录下修改voc_annotation.py
运行
python voc_annotation.py
会生成3个带有根目录标签文件
修改model_data/voc_classes.txt内容
修改train.py文件
运行train.py