Individual skill
huggingface-vision-trainer
Trains and fine-tunes vision models for object detection (D-FINE, RT-DETR v2, DETR, YOLOS), image classification (timm models — MobileNetV3, MobileViT, ResNet, ViT/DINOv3 — plus any Transformers classifier), and SAM/SAM2 segmentation using Hugging Face Transformers on Hugging Face Jobs cloud GPUs. Covers COCO-format dataset preparation, Albumentations augmentation, mAP/mAR evaluation, accuracy metrics, SAM segmentation with bbox/point prompts, DiceCE loss, hardware selection, cost estimation, Trackio monitoring, and Hub persistence. Use when users mention training object detection, image classification, SAM, SAM2, segmentation, image matting, DETR, D-FINE, RT-DETR, ViT, timm, MobileNet, ResNet, bounding box models, or fine-tuning vision models on Hugging Face Jobs.
skills/huggingface-vision-trainer/SKILL.md
Source commit 25910b3d0ba31a2ba102f901cdc9e008c278a69c
What was checked
- Matches the observed repository HEAD
- Yes
- Scanner rules are current
- Yes
- Repository scan complete
- No
- Skill scan complete
- No
- Observed behavior
- Network or broad shell
- Last checked
- Nov 9, 2026
These are observations at the recorded commit, not a guarantee of safety. Review the source before running it.
Install this skill
Choose an agent in your Stack or copy a command below. Commands fetch the pinned commit and copy only this skill directory. Existing destinations are refused. Copying a command does not install anything.
An install command is unavailable: current rules, complete repository and skill scans, an eligible behavior level and a permitted license are required.
Repository context
- Repository stars (not skill ratings)
- 11,307
- Repository last updated
- Oct 30, 2026
- License
- Apache-2.0
Observed agent integration: No specific integration evidence. Standard SKILL.md installation does not imply verified compatibility.
- engineering Derived category
Files and excerpt
File metadata and permitted, redacted excerpts come from the recorded scan. Source links are pinned to that commit.
Browse files
SKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesSKILL.mdcore · 30,195 bytesreferences/finetune_sam2_trainer.mddoc · 6,432 bytesreferences/hub_saving.mddoc · 16,949 bytesreferences/image_classification_training_notebook.mddoc · 11,075 bytesreferences/object_detection_training_notebook.mddoc · 29,917 bytesreferences/reliability_principles.mddoc · 9,487 bytesreferences/timm_trainer.mddoc · 3,547 bytesscripts/dataset_inspector.pyruns · 31,851 bytesscripts/estimate_cost.pyruns · 7,400 bytesscripts/image_classification_training.pyruns · 13,385 bytesscripts/object_detection_training.pyruns · 27,371 bytesscripts/sam_segmentation_training.pyruns · 13,946 bytes
Read excerpt
# 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