当前位置:网站首页>Yolov3 trains its own data set (mmdetection)
Yolov3 trains its own data set (mmdetection)
2022-07-02 07:42:00 【chenf0】
use FasterRcnn Trained the data set marked by myself Voc Format , Now I want to use yolo Let's train , Revised yolo The contents of the document , I'm going to use it directly yolo Training voc Formatted data , There's a problem , Because I'm in a hurry , There is no further detailed study .
MMDetection Most of the training models are coco Format design , Plan to put voc Format conversion to coco Format , It is also convenient for the training of other models in the future .
1.voc Format conversion to coco Format
May refer to :
https://github.com/Stephenfang51/VOC_to_COCO
2.coco Modification of relevant documents
May refer to
【mmdetection】 Use custom coco Format data set for training and testing
(1) Define data types (mmdetection/mmdet/datasets/coco.py), hold CLASSES the tuple Change to the category corresponding to your data set tuple that will do .
CLASSES = ('Other Car', 'Taxi')
(2) modify coco_classes Dataset categories (mmdetection/mmdet/core/evaluation/class_names.py)
def coco_classes():
return [
'Other Car', 'Taxi'
]
(3) Modify usage model model In the dictionary num_classes
num_classes=2,# Number of categories
3. Training
python tools/train.py configs/yolo/yolov3_d53_mstrain-608_273e_coco.py
边栏推荐
- 【信息检索导论】第二章 词项词典与倒排记录表
- [tricks] whiteningbert: an easy unsupervised sentence embedding approach
- Open failed: enoent (no such file or directory) / (operation not permitted)
- Faster-ILOD、maskrcnn_benchmark安装过程及遇到问题
- 传统目标检测笔记1__ Viola Jones
- Drawing mechanism of view (II)
- 【BERT,GPT+KG调研】Pretrain model融合knowledge的论文集锦
- 【Sparse-to-Dense】《Sparse-to-Dense:Depth Prediction from Sparse Depth Samples and a Single Image》
- Thesis writing tip2
- 【Mixed Pooling】《Mixed Pooling for Convolutional Neural Networks》
猜你喜欢

【Paper Reading】

【信息检索导论】第二章 词项词典与倒排记录表

Record of problems in the construction process of IOD and detectron2

Faster-ILOD、maskrcnn_benchmark训练coco数据集及问题汇总

论文写作tip2

【Hide-and-Seek】《Hide-and-Seek: A Data Augmentation Technique for Weakly-Supervised Localization xxx》

【Wing Loss】《Wing Loss for Robust Facial Landmark Localisation with Convolutional Neural Networks》

Proof and understanding of pointnet principle

【多模态】CLIP模型

label propagation 标签传播
随机推荐
parser.parse_args 布尔值类型将False解析为True
【Programming】
Using MATLAB to realize: power method, inverse power method (origin displacement)
【Sparse-to-Dense】《Sparse-to-Dense:Depth Prediction from Sparse Depth Samples and a Single Image》
Feeling after reading "agile and tidy way: return to origin"
Win10+vs2017+denseflow compilation
半监督之mixmatch
【信息检索导论】第六章 词项权重及向量空间模型
Drawing mechanism of view (I)
win10+vs2017+denseflow编译
[paper introduction] r-drop: regulated dropout for neural networks
Memory model of program
Faster-ILOD、maskrcnn_ Benchmark training coco data set and problem summary
[introduction to information retrieval] Chapter 1 Boolean retrieval
Thesis writing tip2
常见的机器学习相关评价指标
Three principles of architecture design
论文tips
Use Baidu network disk to upload data to the server
TimeCLR: A self-supervised contrastive learning framework for univariate time series representation