MMDetection模型微调
2021/10/27 23:11:51
本文主要是介绍MMDetection模型微调,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
使用FasterRCNN训练模型时,因为做的是交通场景下的出租车识别,自己标注了一部分数据,为增强效果,先在数据集BDD100K上训练,然后在自己数据集上进行微调。
使用·faster_rcnn_r50_fpn_1x进行训练
BDD100K数据集下载地址:https://doc.bdd100k.com/download.html
模型下载地址:https://github.com/SysCV/bdd100k-models
通过查看文档,官方提供的训练好的模型是与MMDetection相匹配的,不用自己训练了,直接下载使用即可。
faster_rcnn_r50_fpn_1x对应的模型下载地址为:
https://dl.cv.ethz.ch/bdd100k/det/models/faster_rcnn_r50_fpn_1x_det_bdd100k.pth
修改MMDetection代码,数据集等的修改以及常见问题可参考
MMDetection使用及相关问题
与直接训练相比,微调只需要修改configs/base/default_runtime.py中load_from = None
,将None改为下载模型的路径,或者直接将上面下载地址复制
load_from = "work_dirs/faster_rcnn_r50_fpn_1x_det_bdd100k.pth"
其他与训练自己的数据集一样操作。
MMDetection微调文档参考docs/tutorials/finetune.md
https://github.com/open-mmlab/mmdetection/blob/master/docs/tutorials/finetune.md
这篇关于MMDetection模型微调的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!
- 2024-11-13Slicm 框架怎么进行用户认证?-icode9专业技术文章分享
- 2024-11-13在查询时将 map_coord 列的值转换为字符串有哪些方法?-icode9专业技术文章分享
- 2024-11-13如何将微信地区改成自定义文案?-icode9专业技术文章分享
- 2024-11-13DNS 缓存存在问题有哪些症状和解决方法?-icode9专业技术文章分享
- 2024-11-13HTTP 状态码(405)-Method Not Allowed是什么意思?-icode9专业技术文章分享
- 2024-11-13HTTP 状态码(500)-Internal Server Error是什么意思?-icode9专业技术文章分享
- 2024-11-13在 Element UI 中无法修改 $confirm 的取消按钮文字是什么原因?-icode9专业技术文章分享
- 2024-11-13unity XR是什么?-icode9专业技术文章分享
- 2024-11-13伴随矩阵是什么?-icode9专业技术文章分享
- 2024-11-13怎么使用grep -E 来查找匹配最后 2 条数据?-icode9专业技术文章分享