search

hustvl / YOLOS

[NeurIPS 2021] You Only Look at One Sequence
https://arxiv.org/abs/2106.00666
MIT License
827 stars 118 forks source link

How to train (to detect my custom objects) #2

Closed hujunchao closed 3 years ago

hujunchao commented 3 years ago

❔Question

Thank you for your great work! And I have a question that how to train and to detect my custom objects.

Additional context

Yuxin-CV commented 3 years ago

Hi @hujunchao, thanks for your interest in our work!

This project is based on DETR, we suggest you refer to custom dataset related issues in the DETR repo, e.g., https://github.com/facebookresearch/detr/issues/9.

Yuxin-CV commented 3 years ago

We believe we have answered your question, and as such I'm closing this issue, but let us know if you have further questions.