mmdetection 用法随笔

open-mmlab/mmdetection: OpenMMLab Detection Toolbox and Benchmark是Open-MMLab推出的开箱即用目标检测工具箱，功能非常强大，简单易用，适合迅速从入门到科研。

启动

GPU

单GPU

python tools/train.py ${config_file_path} [--gpu-id ${gpu-no}]

多GPU

bash tools/dis_train.sh ${config_file_path} ${gpu-num}

指定GPU

CUDA_VISIBLE_DEVICES后加GPU_ID，gpu id见nvidia-smi命令给出的id

CUDA_VISIBLE_DEVICES=${GPU_IDS likes 1,2} bash tools/dis_train.sh ${config_file_path} ${gpu-num}

训练

从checkpoins恢复

启动命令后附加 --resume ${path path}

bash tools/dis_train.sh ${config_file_path} ${gpu-num}

未完待续

......