SteveImmanuel / SegGPT-FineTune

Fine-tune SegGPT model with custom datasets
MIT License
24 stars 2 forks source link

checkpoint file #5

Closed xiaotiancai899 closed 1 month ago

xiaotiancai899 commented 2 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 1 month 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