Skip to content

Fine-tuning SAMs for class-aware computer vision tasks in specific scenarios

License

Notifications You must be signed in to change notification settings

wmpscc/finetune-anything

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

56 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

finetune-anything

Introduction

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.

The complete code is coming soon.

Install

git clone https://github.com/ziqi-jin/finetune-anything.git
cd finetune-anything
pip install -r requirements.txt
the complete code is comming soon.

Supported Tasks

Train

Test

Deploy

  • Onnx export

About

Fine-tuning SAMs for class-aware computer vision tasks in specific scenarios

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%