@huggingface/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.

View in AI SkillSafe app
Scanned · no findings
0 downloads
0 stars
0 demos
SKILL.md

Embed badges

Add these to your README to show the skill's verification status.

SkillSafe verified badge
Verified badge
[![SkillSafe verified badge](https://api.skillsafe.ai/v1/badge/@huggingface/huggingface-vision-trainer/verified)](https://skillsafe.ai/skill/@huggingface/huggingface-vision-trainer/)
Installs badge
Installs badge
[![Installs badge](https://api.skillsafe.ai/v1/badge/@huggingface/huggingface-vision-trainer/installs)](https://skillsafe.ai/skill/@huggingface/huggingface-vision-trainer/)
Scan badge
Scan badge
[![Scan badge](https://api.skillsafe.ai/v1/badge/@huggingface/huggingface-vision-trainer/scan)](https://skillsafe.ai/skill/@huggingface/huggingface-vision-trainer/)
Eval pass rate badge
Eval pass rate
[![Eval pass rate badge](https://api.skillsafe.ai/v1/badge/@huggingface/huggingface-vision-trainer/eval)](https://skillsafe.ai/skill/@huggingface/huggingface-vision-trainer/)