facebook

/

detr-resnet-50

like 658

Just use a MPL Classifier I use YOLO v.whatever

Hi there, here are some useful resources on how to fine-tune DETR:
https://huggingface.co/docs/transformers/main/en/model_doc/detr#resources

I want to know if It is possible train the model over the pretrained model and how to do it.

To fine tune this model, I personally used the Jupyter Notebook at How to Train DETR with 🤗 Transformers on a Custom Dataset as a guide.

Also would be desiderable to know how my dataset have to be labeled in order to feed the model.

There are many ways to label your dataset. The approach I've taken is to use Label Studio, an open-source solution for labeling data collaboratively. You can export the labels in whatever format suits you. COCO works best for the Notebook I've shared.

I want to know if It is possible train the model over the pretrained model and how to do it.

To fine tune this model, I personally used the Jupyter Notebook at How to Train DETR with 🤗 Transformers on a Custom Dataset as a guide.

Also would be desiderable to know how my dataset have to be labeled in order to feed the model.

There are many ways to label your dataset. The approach I've taken is to use Label Studio, an open-source solution for labeling data collaboratively. You can export the labels in whatever format suits you. COCO works best for the Notebook I've shared.

Thanks for providing the Jupyter Notebook link. It is nice to run code on cloud. But if the dataset is too large, Google doesn't allow long-time training.
Is it possible to just copy-past all the codes down to local PC and train the model according to personal needs?

Is it possible to just copy-past all the codes down to local PC and train the model according to personal needs?

Yup! That's what I did.