当前位置:网站首页>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
边栏推荐
- Calculate the difference in days, months, and years between two dates in PHP
- 使用Matlab实现:幂法、反幂法(原点位移)
- Alpha Beta Pruning in Adversarial Search
- Huawei machine test questions-20190417
- Point cloud data understanding (step 3 of pointnet Implementation)
- Use matlab to realize: chord cut method, dichotomy, CG method, find zero point and solve equation
- One field in thinkphp5 corresponds to multiple fuzzy queries
- 【Ranking】Pre-trained Language Model based Ranking in Baidu Search
- Using compose to realize visible scrollbar
- Oracle RMAN automatic recovery script (migration of production data to test)
猜你喜欢
一份Slide两张表格带你快速了解目标检测
【深度学习系列(八)】:Transoform原理及实战之原理篇
Illustration of etcd access in kubernetes
[introduction to information retrieval] Chapter 1 Boolean retrieval
MySQL has no collation factor of order by
Tencent machine test questions
Play online games with mame32k
Point cloud data understanding (step 3 of pointnet Implementation)
Using MATLAB to realize: Jacobi, Gauss Seidel iteration
[model distillation] tinybert: distilling Bert for natural language understanding
随机推荐
Oracle EBS database monitoring -zabbix+zabbix-agent2+orabbix
Optimization method: meaning of common mathematical symbols
华为机试题-20190417
CRP implementation methodology
ERNIE1.0 与 ERNIE2.0 论文解读
ORACLE APEX 21.2安裝及一鍵部署
Oracle EBS DataGuard setup
使用Matlab实现:弦截法、二分法、CG法,求零点、解方程
机器学习理论学习:感知机
CSRF attack
離線數倉和bi開發的實踐和思考
Drawing mechanism of view (I)
Alpha Beta Pruning in Adversarial Search
JSP intelligent community property management system
[introduction to information retrieval] Chapter 7 scoring calculation in search system
PointNet理解(PointNet实现第4步)
类加载器及双亲委派机制
Oracle RMAN automatic recovery script (migration of production data to test)
【Torch】最简洁logging使用指南
Data warehouse model fact table model design