Downloads · 30 days
0
TrainingCat/car-paint-defect-detection
car-paint-defect-detection is a image classification model from TrainingCat. Use it when you need a label for an image. The card lists the license as cc-by-4.0.
Downloads · 30 days
0
Access
Public
Updated Dec 3, 2025
Repo size
1.4 GB
Likes
0
Public
Click a slice to open those files.
.pth1.4 GB · 100%
From the Hugging Face model README
基于深度学习的汽车漆面缺陷检测系统,包含三个核心模块。
| 模型 | 文件 | 大小 | 用途 |
|---|---|---|---|
| DQN图像增强 | task1_dqn/dqn_final.pth | ~987MB | 强化学习图像增强 |
| UNet++分割 | task2_unetpp/best_model.pth | ~105MB | 缺陷区域分割 (IoU=0.884) |
| ResNet分类 | task3_classification/best_model.pth | ~283MB | 缺陷类型分类 (Acc=96.3%) |
from huggingface_hub import hf_hub_download
# 下载分割模型
model_path = hf_hub_download(
repo_id="TrainingCat/car-paint-defect-detection",
filename="task2_unetpp/best_model.pth"
)
或使用命令行:
huggingface-cli download TrainingCat/car-paint-defect-detection --local-dir models/