当前位置:网站首页>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
边栏推荐
- Practice and thinking of offline data warehouse and Bi development
- 【深度学习系列(八)】:Transoform原理及实战之原理篇
- 【Sparse-to-Dense】《Sparse-to-Dense:Depth Prediction from Sparse Depth Samples and a Single Image》
- Alpha Beta Pruning in Adversarial Search
- [torch] the most concise logging User Guide
- MySQL composite index with or without ID
- SSM personnel management system
- Play online games with mame32k
- Delete the contents under the specified folder in PHP
- Typeerror in allenlp: object of type tensor is not JSON serializable error
猜你喜欢
Faster-ILOD、maskrcnn_ Benchmark trains its own VOC data set and problem summary
[Bert, gpt+kg research] collection of papers on the integration of Pretrain model with knowledge
ABM thesis translation
[introduction to information retrieval] Chapter 7 scoring calculation in search system
【Random Erasing】《Random Erasing Data Augmentation》
TimeCLR: A self-supervised contrastive learning framework for univariate time series representation
【MEDICAL】Attend to Medical Ontologies: Content Selection for Clinical Abstractive Summarization
Use Baidu network disk to upload data to the server
【Ranking】Pre-trained Language Model based Ranking in Baidu Search
MMDetection安装问题
随机推荐
Faster-ILOD、maskrcnn_benchmark训练自己的voc数据集及问题汇总
【多模态】CLIP模型
win10+vs2017+denseflow编译
Tencent machine test questions
Semi supervised mixpatch
Label propagation
Transform the tree structure into array in PHP (flatten the tree structure and keep the sorting of upper and lower levels)
常见CNN网络创新点
论文写作tip2
【FastDepth】《FastDepth:Fast Monocular Depth Estimation on Embedded Systems》
Determine whether the version number is continuous in PHP
Latex formula normal and italic
A slide with two tables will help you quickly understand the target detection
Generate random 6-bit invitation code in PHP
Drawing mechanism of view (3)
label propagation 标签传播
【Wing Loss】《Wing Loss for Robust Facial Landmark Localisation with Convolutional Neural Networks》
Implementation of purchase, sales and inventory system with ssm+mysql
Machine learning theory learning: perceptron
【论文介绍】R-Drop: Regularized Dropout for Neural Networks