Communitygithub.com

dhar174/hugging-face-vision-trainer

使用 Hugging Face Transformers 在 Hugging Face Jobs 上訓練和微調視覺模型,支援物件偵測、影像分類和 SAM/SAM2 分割。

hugging-face-vision-trainer 是什麼?

此技能專為開發者和 AI 代理設計,提供一個統一的介面,用於在 Hugging Face 平台上訓練和微調各種視覺模型。它支援多種電腦視覺任務:物件偵測(包括 D-FINE、RT-DETR v2、DETR 和 YOLOS)、影像分類(支援 timm 模型如 MobileNetV3、MobileViT、ResNet、ViT/DINOv3 以及任何 Transformers 分類器)以及 SAM/SAM2 分割。透過整合 Hugging Face Jobs,此技能可讓使用者無需手動管理基礎架構,即可大規模啟動訓練工作。其核心價值在於簡化設定流程、提供可自訂的訓練參數(如學習率、批次大小和時期數),並自動處理資料集轉換和模型保存。此技能可在支援 CLI 和 API 呼叫的 AI 代理平台(如 Claude Code、Cursor 或 Codex)中順暢運作,對於需要快速迭代視覺模型的機器學習工程師和研究人員來說是不可或缺的工具。

相容平台~Claude Code~Codex CLI~Cursor
npx skills add https://github.com/dhar174/DeskBuddy/tree/main/.agents/skills/hugging-face-vision-trainer

在你喜歡的 AI 中提問

開啟一個已預先載入此 Agent Skill 的新對話。

說明文件

hugging-face-vision-trainer 是做什麼的?

使用 Hugging Face Transformers 在 Hugging Face Jobs 上訓練和微調視覺模型,支援物件偵測 (D-FINE、RT-DETR v2、DETR、YOLOS)、影像分類 (timm 模型 — MobileNetV3、MobileViT、ResNet、ViT/DINOv3 — 以及任何 Transformers 分類器) 和 SAM/SAM2 分割。

Individual skills in this repo

This repo contains 2 individual skills — each has its own dedicated page.

相關技能