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.

Target: Design & content creation. Automatically collected · unverified.

Excerpt from the author’s documentation

Extracted from a fixed version of SKILL.md in its original language. This is the author’s documentation; installation, compatibility and runtime results have not been verified by this site.

Read the original English text

Topics in the source

  • Vision Model Training on Hugging Face Jobs
  • When to Use This Skill
  • Related Skills
  • Local Script Execution
  • Prerequisites Checklist
  • Account & Authentication
  • Dataset Requirements — Object Detection
  • Dataset Requirements — Image Classification

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. 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 hugging-face-jobs — General HF Jobs infrastructure: token authentication, hardware flavors, timeout management, cost estimation, secrets, environment variables, scheduled jobs, and result persistence. Refer to the Jobs skill for any non-training-specific Jobs questions (e.g., "how do secrets work?", "what hardware is available?", "how do I pass tokens?"). hugging-face-model

This is an excerpt. Refer to the complete source and its current requirements.

Source repository
huggingface/skills
Source file
skills/huggingface-vision-trainer/SKILL.md
Fixed revision
25910b3d0ba31a2ba102f901cdc9e008c278a69c
Read the complete original documentation ↗

Sources and evidence

This record was collected automatically. Its documentation, suitability and runtime behavior have not been manually verified.

Author: huggingface/skills · original repository metadata. Documentation review: Not reviewed. Last source check: 2026-11-09T16:01:02.755Z.

Categories are inferred from names, descriptions and paths. They do not establish compatibility.

License and attribution

Apache-2.0. Third-party content and copyright remain with their authors. Refer to the original license and terms.

Original license ↗

Synchronization, copying and user-reported success do not mark a resource as runtime-tested.

Limitations and notes

  • Collecting metadata does not mean this site has installed, adapted, runtime-tested or security-reviewed the resource.