[极简]nanodet训练自己的数据集

一、准备数据集

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