mmdetection常用命令

本篇博客是本人在进行目标检测实验中使用到的一些命令,不定时更新。

当我们自定义了一个模型,需要查看模型的参数量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

  • 6
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值