The Segment Anything Model (SAM) has revolutionized computer vision. Relying on fine-tuning of SAM will solve a large number of basic computer vision tasks. We are designing a class-aware one-stage tool for training fine-tuning models based on SAM.
You need to supply the datasets for your tasks and the supported task name, this tool will help you to get a finetuned model for your task.
Now you can not run the code yet. I am working on it and the complete code is coming soon.
git clone https://github.com/ziqi-jin/finetune-anything.git
cd finetune-anything
pip install -r requirements.txt
the complete code is comming soon.
- Onnx export