SteveImmanuel / SegGPT-FineTune

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

checkpoint file #5

Closed xiaotiancai899 closed 3 months ago

xiaotiancai899 commented 4 months ago

Would you please upload the tuning.pt file mentioned in the inference.py file? As a result, I can use the trained and tuned SegGPT model to predict other images. Thanks a lot!!

SteveImmanuel commented 3 months ago

Please refer to my other repo specifically for that purpose: https://github.com/SteveImmanuel/OEM-Few-Shot-Learnable-Prompt You can find the checkpoint in the release