train istr with custom dataset

hujiecpp / ISTR

ISTR: End-to-End Instance Segmentation with Transformers (https://arxiv.org/abs/2105.00637)

203 stars 28 forks source link

Open imenselmi opened 2 years ago

imenselmi commented 2 years ago

how can i train istr with custom dataset ?

imenselmi commented 2 years ago

which files should i change?

imenselmi commented 2 years ago

also Is it possible to use an ISTR with custom dataset or not ?

hujiecpp commented 2 years ago

Yes, please kindly refer to the doc of Detectron2, which introduces how to implement custom dataset to train your own models.