← all repositories

ziqi-jin/finetune-anything

A fine-tuning framework for the Segment Anything Model targeting semantic segmentation, matting, and detection tasks.

864 stars Python Computer VisionML Frameworks
finetune-anything
Velocity · 7d
+0.7
★ / day
Trend
steady
star history

This tool encapsulates the three core components of SAM — Image Encoder Adapter, Prompt Encoder Adapter, and Mask Decoder Adapter — to enable task-specific fine-tuning. Users supply datasets and specify supported tasks to obtain trained models, with configurable learning rates and weight decay for unfrozen parameters. The framework supports semantic segmentation, matting, instance segmentation, and detection tasks with corresponding dataset loaders.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.