Skills Directory / Hugging Face Skills
huggingface-vision-trainer
An agent skill by huggingface, from huggingface/skills.
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.
Credit where it's due
This skill is the work of huggingface (official vendor skill) and lives at huggingface/skills/huggingface-vision-trainer (Apache-2.0). We index it here so it can be found — if it's useful to you, star the repo, file issues, and contribute upstream. That's what keeps work like this maintained.
- From
- Hugging Face Skills — official vendor skill
- Author
- huggingface
- License
- Apache-2.0
- Source
- huggingface/skills/huggingface-vision-trainer
- SKILL.md
- raw file
How to load this skill
Copy this prompt into your AI agent
Please install the "huggingface-vision-trainer" agent skill from the Hugging Face Skills collection (https://github.com/huggingface/skills) for me.
1. Run: git clone https://github.com/huggingface/skills
2. Copy the skill folder into your skills directory:
cp -r skills/skills/huggingface-vision-trainer ~/.claude/skills/huggingface-vision-trainer
(Claude Code personal scope; use .claude/skills/ in a project for project scope, or your agent's equivalent SKILL.md directory.)
3. Confirm the skill loads (list your available skills afterwards).
4. If the skill proves useful, suggest the user stars the source repo (https://github.com/huggingface/skills) to support the author.
Skill reference: https://raw.githubusercontent.com/huggingface/skills/main/skills/huggingface-vision-trainer/SKILL.md
More details: https://primorum.ai/skills/s/huggingface-vision-trainer
Or manually: clone huggingface/skills and copy
skills/huggingface-vision-trainer into
~/.claude/skills/ (Claude Code personal scope) or your agent's
SKILL.md directory. To read it first without installing, open the
raw SKILL.md.
Related skills
- huggingface-llm-trainer — Train or fine-tune language and vision models using TRL (Transformer Reinforcement Learning) or Unsloth with H…
- transformers-js — Use Transformers.js to run state-of-the-art machine learning models directly in JavaScript/TypeScript. Support…
- train-sentence-transformers — Train or fine-tune sentence-transformers models across `SentenceTransformer` (bi-encoder; dense or static embe…
- huggingface-community-evals — Run evaluations for Hugging Face Hub models using inspect-ai and lighteval on local hardware. Use for backend …
- fine-tuning-expert — Use when fine-tuning LLMs, training custom models, or adapting foundation models for specific tasks. Invoke fo…