一、准备数据集

cd nanodet
mkdir demo
cd demo
mkdir train
cd train
mkdir img
mkdir ann
cd ..
mkdir val
cd val
mkdir img
mkdir ann
cd ..

把原图片和xml标注文件放到train和val下面

二、修改配置文件

cp ./config/nanodet_custom_xml_dataset.yml ./demo

1、修改save_dir
2、修改num_classes
3、修改class_names
4、修改4个path

三、开始训练

python tools/train.py CONFIG_FILE_PATH

四、可视化

    cd <YOUR_SAVE_DIR>
    tensorboard --logdir ./

五、测试

• Inference images

python demo/demo.py image --config CONFIG_PATH --model MODEL_PATH --path IMAGE_PATH

• Inference video

python demo/demo.py video --config CONFIG_PATH --model MODEL_PATH --path VIDEO_PATH

• Inference webcam

python demo/demo.py webcam --config CONFIG_PATH --model MODEL_PATH --camid YOUR_CAMERA_ID