search

SteveImmanuel / SegGPT-FineTune

Fine-tune SegGPT model with custom datasets
MIT License
27 stars 3 forks source link

OEM dataset #4

Closed xiaotiancai899 closed 3 months ago

xiaotiancai899 commented 4 months ago

Thanks for your good work! As you mentioned, you used the OEM dataset. Could you please put the dataset folder into the model folder so that make the whole project more complete?

SteveImmanuel commented 3 months ago

This repo I specifically develop to serve as general purpose template for fine-tuning so I am not going to commit to such project structure. However, you can easily override the dataloader to support your own custom dataset.