wmpscc / finetune-anything

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

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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

License:MIT License


Languages

Language:Python 100.0%