本篇博客是本人在进行目标检测实验中使用到的一些命令,不定时更新。
当我们自定义了一个模型,需要查看模型的参数量params和flops时,使用如下命令:
python tools\analysis_tools\get_flops.py ${your config file}
训练命令:
python tools\train.py ${your config file} --work-dir ${your output path} [--resume]
resume将会接着上次的epoch继续训练,直到达到设置的最大epoch
测试命令:
python tools\test.py ${your config file} ${your pth file}
[--show]
[--show-dir your results save path]
使用--show将会在窗口中展示原图和测试后的图像,--show-dir将会把show展示的结果保存到指定路径,这里我使用的是绝对路径,请注意,这两个命令相互独立,可以单独运行。
指定GPU训练/测试:
CUDA_VISIBLE_DEVICES=1 python train.py
即在训练/测试命令前加入该指令
按照间隔数保存模型权重
default_hooks = dict(checkpoint=dict(type='CheckpointHook', interval=5, by_epoch=True))
设置interval的值即可
多卡训练:
CUDA_VISIBLE_DEVICES=0,1,2... tools/dist_train.sh configs ${gpu_nums} --work-dir ${work_dir} --resume