当前位置:网站首页>MMDetection模型微调
MMDetection模型微调
2022-07-02 06:26:00 【chenf0】
使用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代码,数据集等的修改以及常见问题可参考
与直接训练相比,微调只需要修改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
边栏推荐
- 中年人的认知科普
- [introduction to information retrieval] Chapter 7 scoring calculation in search system
- Use matlab to realize: chord cut method, dichotomy, CG method, find zero point and solve equation
- ERNIE1.0 与 ERNIE2.0 论文解读
- Oracle 11g sysaux table space full processing and the difference between move and shrink
- 使用Matlab实现:弦截法、二分法、CG法,求零点、解方程
- Oracle APEX 21.2 installation et déploiement en une seule touche
- Point cloud data understanding (step 3 of pointnet Implementation)
- [tricks] whiteningbert: an easy unsupervised sentence embedding approach
- @Transitional step pit
猜你喜欢
[introduction to information retrieval] Chapter 1 Boolean retrieval
【模型蒸馏】TinyBERT: Distilling BERT for Natural Language Understanding
Three principles of architecture design
常见的机器学习相关评价指标
【信息检索导论】第七章搜索系统中的评分计算
Two table Association of pyspark in idea2020 (field names are the same)
SSM personnel management system
第一个快应用(quickapp)demo
A slide with two tables will help you quickly understand the target detection
MySQL无order by的排序规则因素
随机推荐
ssm垃圾分类管理系统
Message queue fnd in Oracle EBS_ msg_ pub、fnd_ Application of message in pl/sql
ORACLE APEX 21.2安裝及一鍵部署
[introduction to information retrieval] Chapter 3 fault tolerant retrieval
ERNIE1.0 与 ERNIE2.0 论文解读
类加载器及双亲委派机制
华为机试题
Module not found: Error: Can't resolve './$$_gendir/app/app.module.ngfactory'
ABM论文翻译
Oracle EBS ADI development steps
中年人的认知科普
Using MATLAB to realize: Jacobi, Gauss Seidel iteration
PHP returns the corresponding key value according to the value in the two-dimensional array
【Ranking】Pre-trained Language Model based Ranking in Baidu Search
软件开发模式之敏捷开发(scrum)
Typeerror in allenlp: object of type tensor is not JSON serializable error
Use matlab to realize: chord cut method, dichotomy, CG method, find zero point and solve equation
离线数仓和bi开发的实践和思考
Explanation of suffix of Oracle EBS standard table
Oracle 11.2.0.3 handles the problem of continuous growth of sysaux table space without downtime