if i have a few hundred images to do , will it run on small dataset ? I mean can I train it on small dataset? Thank you for your amazing work

[D-Mad]

[acaelles97]

Hi, thanks for your interest! Yes you can train the model on a small dataset as long as you fine-tune the model from YT-21/OVIS depending on which is more similar to yours. You can take a look at https://github.com/acaelles97/DeVIS/blob/master/src/datasets/vis.py to follow instructions on the structure you need for the dataset. To train the model, you can take a look at https://github.com/acaelles97/DeVIS/blob/master/docs/TRAIN.md for instructions.

Let me know if you need more help.