search

j-marple-dev / AYolov2

GNU General Public License v3.0
156 stars 15 forks source link

how to use a yolov5 representation learning model? #54

Closed tangsipeng closed 2 years ago

tangsipeng commented 2 years ago

i do not quite understand how to use a yolov5 representation learning model of coco ? it supposed to be a pretrain model like imagenet pretrained?

JeiKeiLim commented 2 years ago

Thank you for asking. I forgot to mention that our model does not use pre-trained model. Our goal was to make an object detection model without pre-trained weights such as image-net. And representation learning is one way to improve backbone capability with limited amount of data. Representation learning is still under progress and experimental but if you wish to try, you can first train backbone model with train_repr.py and train with train.py as you would train a scratch model but with backbone trained model.

JeiKeiLim commented 2 years ago

I'm closing this. Please open new issue if you have further question.