search

open-mmlab / mmrazor

OpenMMLab Model Compression Toolbox and Benchmark.
https://mmrazor.readthedocs.io/en/latest/
Apache License 2.0
1.47k stars 227 forks source link

How to train my own datasets? #143

Open Muke6 opened 2 years ago

Muke6 commented 2 years ago

I use data in coco format, the category is a category, how to modify the category？ 微信截图_20220418201903

pprp commented 2 years ago

This document might be helpful to you. https://mmdetection.readthedocs.io/en/latest/tutorials/customize_dataset.html#modify-dataset-classes

Audrey528 commented 2 years ago

@pprp I don't alter which files to change my own classes, I need some help. Thanks.

pprp commented 2 years ago

Hi, @Audrey528

Can you provide the full config?

Audrey528 commented 2 years ago

@pprp I wanna train voc format dataset using mmdet， I change two places in cwd_cls_head_gfl_r101_fpn_gfl_r50_fpn_1x_coco.py. 1653709331(1) 1653709383(1)

I have some questions.

  1. I have two classes, which file can I alter classes?
  2. just like the below picture, do I need to change the part cfg? 1653709523(1) 3.How can I install mmdet? 1653709769(1)
Audrey528 commented 2 years ago

@pprp I have solved 1 and 3.

pprp commented 2 years ago

Hi, @Audrey528

As for the second question, I think there is no need to change this part config, because the distiller have nothing to do with the number of classes.

xuhao-anhe commented 2 years ago

@pprp I wanna train voc format dataset using mmdet， I change two places in cwd_cls_head_gfl_r101_fpn_gfl_r50_fpn_1x_coco.py. 1653709331(1) 1653709383(1)

I have some questions.

  1. I have two classes, which file can I alter classes?
  2. just like the below picture, do I need to change the part cfg? 1653709523(1) 3.How can I install mmdet? 1653709769(1)

Hello. How did you install mmdet? After I installed it according to the official tutorial, I still reported an error in the train_mmdet.py file.

pprp commented 2 years ago

Hello. How did you install mmdet? After I installed it according to the official tutorial, I still reported an error in the train_mmdet.py file.

@xuhao-anhe hi, sorry for my late response. I used the following commands: python setup.py install or pip install -e . .

As for the error in the train_mmdet.py file, can you share the detailed error log?

luoxinstart commented 1 year ago

I use data in coco format, the category is a category, how to modify the category？ 微信截图_20220418201903

go cheak doc of mmdetection, it has details on using your own dataset