huggingface-vision-trainer
Trains and fine-tunes object detection, image classification and SAM or SAM2 segmentation models on Hugging Face Jobs cloud GPUs.
- Kind
- Skill
- Works with
- Claude Code
- Source
- Hugging Face Skills @ 25910b3d0ba3
- Path
- skills/huggingface-vision-trainer/SKILL.md
- Jobs
- Implementation
- Topics
- Machine learning
Credit
- Made by
- huggingface
- Original
- github.com/huggingface/skills/blob/25910b3d0ba31a2ba102f901cd…
- License
- Apache-2.0
Listed by Skill Manifest; not made by us.
InstallInstall it by hand, following the source.
hf skills add huggingface-vision-trainer
Install factsScanned 2 Nov 2026
- Adds
- .claude/skills/huggingface-vision-trainer/ (the skill folder)
- 11 file(s) the skill refers to
- Runs scripts
- Yes
- scripts/dataset_inspector.py
- scripts/estimate_cost.py
- shell block in the text
- Uses the network
- Yes
- downloads Python packages
- Needs secrets
- Yes
- HF_TOKEN
- Reads environment variables
- HF_TOKEN
- Source revision
- 25910b3d0ba3
- Licence
- Apache-2.0
- Last activity
- 30 Oct 2026
- Review
- Reviewed
Found by a static scan of the stored text; nothing is run. “None found” is not a guarantee, so read what you install before you run it.
- Only the main file is read; other files in the folder are not scanned
# Vision Model Training on Hugging Face Jobs Train object detection, image classification, and SAM/SAM2 segmentation models on managed cloud GPUs. No local GPU setup required—results are automatically saved to the Hugging Face Hub. ## When to Use This Skill Use this skill when users want to: - Fine-tune object detection models (D-FINE, RT-DETR v2, DETR, YOLOS) on cloud GPUs or local - Fine-tune image classification models (timm: MobileNetV3, MobileViT, ResNet, ViT/DINOv3, or any Transformers classifier) on cloud GPUs or local - Fine-tune SAM or SAM2 models for segmentation / image matting using bbox or point prompts - Train bounding-box detectors on custom datasets - Train image classifiers on custom datasets - Train segmentation models on custom mask datasets with prompts - Run vision training jobs on Hugging Face Jobs infrastructure - Ensure trained vision models are permanently saved to the Hub ## Related Skills …