search

Czm369 / MixPL

Mixed Pseudo Labels for Semi-Supervised Object Detection
Apache License 2.0
57 stars 1 forks source link

How can I train a model based on Custom Dataset? #5

Open JeongHanJun opened 8 months ago

JeongHanJun commented 8 months ago

Hello, thank you very much for developing a model with good performance and sharing the contents on GitHub.

We understand that the model performs well. Accordingly, I would like to train a custom dataset.

My current situation is that I created an image based on the DockerFile in the docker folder and created a container.

Now I want to train a Custom Dataset in this container. This Dataset is in coco format.

First of all, I don't know how I should structure the directory structure.

Also, I don't know what code to use to learn the model.

I'd like some advice on this, and if there's anyone who has successfully run a Custom Train, I would greatly appreciate your help!

Czm369 commented 8 months ago

You can refer to https://mmdetection.readthedocs.io/en/latest/user_guides/semi_det.html

vignatej commented 7 months ago

Hello @JeongHanJun and @Czm369 , if you have trained it on custom dataset please send me the code. If you have jupyter notebook, then please send it. Else please send entire code. please comment it here or mail me at: vignatej123@gmail.com. your help will be invaluable.

JeongHanJun commented 7 months ago

Hello @vignatej I can't find a code about how to train custom dataset. I want to train my custom dataset too... If you find how to train, please explain to me please...

Czm369 commented 7 months ago

You can process the custom dataset into the COCO format supported by MMDet for training. https://github.com/Czm369/MixPL/blob/410f044972a427b67f34393fc8c9314e04add8a2/mmdet/datasets/coco.py#L14

BruceLxw commented 4 months ago

Sincerely thank you for making your source code public. I would like to run my own dataset on your code, but now I am encountering an issue where I have also built it using the official semi supervised dataset, following the dataset directory in your config. However, this issue occurred during training: image

I just don't know where to troubleshoot the problem. Have you encountered it before, or could you provide some solutions? Thank you very much

BruceLxw commented 4 months ago

i have solved it !!

vignatej commented 4 months ago

please post here how???

bharanibala commented 3 months ago

i have solved it !!

Hi,

Could you please post the solution on how you made the code work with your custom dataset?

Thanks, Bharani.

bharanibala commented 2 months ago

i have solved it !!

Hello,

I am facing the same issue and tried (https://github.com/open-mmlab/mmdetection/issues/3628) but not able fix it. Could you please share your thoughts!

Thanks, Bharani.