hustvl / YOLOS

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

Control the patches #17

Open loucif01 opened 2 years ago

loucif01 commented 2 years ago

❔Question

Hello, thank you for this great contribution. I'm asking if with your architecture, we can control witch patches to feed to YOLOS, because i have already the RoI (Region of interest) of each image of my dataset, and i want to train the model just on theses regions of image, so can we do this by controlling the patches?

Thanks.

Additional context