Claude Marketplaces
huggingface avatar

huggingface-vision-trainer

huggingface/skillsOfficial

Train and fine-tune object detection models (RTDETRv2, YOLOS, DETR and others) and image classification models (timm and transformers models — MobileNetV3…

GitHub stars
10.1K
View on GitHub

Install

/plugin install huggingface-vision-trainer@huggingface-skills

Run it inside Claude Code after adding the marketplace with /plugin marketplace add huggingface/skills.

Sponsors

CodeRabbitSponsoredTry For FreeLid closed, agents workingDownload freeDevtools for design engineersTry it livebelt - the only tool your agent needsSponsoredone prompt installOpen StepsSponsoredDownload for freeAppSignalSponsoredStart Free TrialBlock distraction from your iPhoneBlock now (100% free)Granola, the best AI meeting recorderSponsoredDownload for free

What it does

Train and fine-tune object detection models (RTDETRv2, YOLOS, DETR and others) and image classification models (timm and transformers models — MobileNetV3, MobileViT, ResNet, ViT/DINOv3) using Transformers Trainer API on Hugging Face Jobs infrastructure or locally. Includes COCO dataset format support, Albumentations augmentation, mAP/mAR metrics, trackio tracking, hardware selection, and Hub persistence.

Contents

Discovered from source

Agents 7

Details

Marketplace
huggingface/skills
Source
skills/huggingface-vision-trainer
Category
community

More from huggingface/skills

All 12