当前位置:网站首页>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
边栏推荐
- Memory model of program
- How do vision transformer work? [interpretation of the paper]
- Calculate the total in the tree structure data in PHP
- Installation and use of image data crawling tool Image Downloader
- 【FastDepth】《FastDepth:Fast Monocular Depth Estimation on Embedded Systems》
- Win10+vs2017+denseflow compilation
- How to efficiently develop a wechat applet
- [torch] the most concise logging User Guide
- MoCO ——Momentum Contrast for Unsupervised Visual Representation Learning
- 常见的机器学习相关评价指标
猜你喜欢
![[in depth learning series (8)]: principles of transform and actual combat](/img/2e/89920de2273b6f1bc3b21a19c2ecbe.png)
[in depth learning series (8)]: principles of transform and actual combat

【Cutout】《Improved Regularization of Convolutional Neural Networks with Cutout》

程序的内存模型

《Handwritten Mathematical Expression Recognition with Bidirectionally Trained Transformer》论文翻译

ModuleNotFoundError: No module named ‘pytest‘

自然辩证辨析题整理

【Ranking】Pre-trained Language Model based Ranking in Baidu Search

SSM laboratory equipment management
![[introduction to information retrieval] Chapter 6 term weight and vector space model](/img/42/bc54da40a878198118648291e2e762.png)
[introduction to information retrieval] Chapter 6 term weight and vector space model

【信息检索导论】第六章 词项权重及向量空间模型
随机推荐
使用百度网盘上传数据到服务器上
Conversion of numerical amount into capital figures in PHP
The difference and understanding between generative model and discriminant model
【Paper Reading】
Point cloud data understanding (step 3 of pointnet Implementation)
Common machine learning related evaluation indicators
Machine learning theory learning: perceptron
机器学习理论学习:感知机
Two dimensional array de duplication in PHP
ModuleNotFoundError: No module named ‘pytest‘
Sorting out dialectics of nature
【TCDCN】《Facial landmark detection by deep multi-task learning》
SSM laboratory equipment management
[introduction to information retrieval] Chapter 6 term weight and vector space model
【MEDICAL】Attend to Medical Ontologies: Content Selection for Clinical Abstractive Summarization
【Mixup】《Mixup:Beyond Empirical Risk Minimization》
Semi supervised mixpatch
【多模态】CLIP模型
【Programming】
Huawei machine test questions-20190417