mmdetection试水

之前没用过这种工具,一看代码好多就望而生畏了,这两天突发奇想研究一下,感觉突然发现了新大陆!!!这也太方便了吧!!!什么都集成好了,写个config文件配个模块就能完成自己的网络啦

ps.看到人家8个16个gpus,分布式训练羡慕。。。

mmdetection的主要网络结构编写放在config文件中,需要使用什么在config文件中配就行。

刚入坑可以看看:https://zhuanlan.zhihu.com/p/101225733

接下来就是一些小tricks

1.如何测试模型获得各个指标

python tools/test.py ${CONFIG_FILE} ${CHECKPOINT_FILE} [--out ${RESULT_FILE}] [--eval ${EVAL_METRICS}] [--show]
python tools/coco_eval.py ./demo/result.json --an 
 ./data/coco/annotations/instances_val2017.json --types segm

通过test生成json文件,然后通过json文件使用coco_eval.py测试各个指标。

注:(1)这里可能会报错。

    result_file = result_files[res_type]
    TypeError: string indices must be integers

解决方法:将mmdet/core/evaluation/coco_utils.py文件中上面一句代码替换为下面一句。

 # result_file = result_files[res_type]
        result_file = result_files

(2)如果报'Results do not correspond to current coco set',则一般是测试集anns文件路径错误。

2.模型训练

我这里训练了polarmask 4gpus 中的configs文件,注意只需要更改一下学习率。

更改原则:

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值