一、安装环境
# install requirements
pip install pycocotools numpy opencv-python tqdm tensorboard tensorboardX pyyaml
pip install torch==1.4.0pip install torchvision==0.5.0
二、下载pytorch版efficientdet源码
git clone https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch.git
源码链接:https://github.com/zylo117/Yet-Another-EfficientDet-Pytorch
三、准备数据集
# your dataset structure should be like thisdatasets/
-your_project_name/
-train_set_name/
-*.jpg-val_set_name/
-*.jpg-annotations-instances_{train_set_name}.json-instances_{val_set_name}.json
#forexample, coco2017
datasets/
-coco2017/
-train2017/
-000000000001.jpg-000000000002.jpg-000000000003.jpg-val2017/
-000000000004.jpg-000000000005.jpg-000000000006.jpg-annotations-instances_train2017.json-instances_val2017.json
四、修改配置文件</