ONNX 模型库
返回模型

说明文档

焊接缺陷检测模型 (YOLOv8)

这是一个经过微调的 YOLOv8 模型,用于检测以下焊接缺陷:

  • 合格焊缝 (Good Weld)
  • 不合格焊缝 (Bad Weld)
  • 缺陷 (Defect)

包含完整的训练产物:

  • 最佳模型权重 (best.pt)
  • 最终模型权重 (last.pt)
  • 训练日志
  • 验证预测结果
  • 增强训练批次

训练详情

  • 模型类型:YOLOv8
  • 数据集:The Welding Defect Dataset - v2
  • 类别数:3
  • 训练轮数:100
  • 图像尺寸:640x640
  • 批次大小:16

使用方法

from ultralytics import YOLO

model = YOLO("YOUR_HF_USERNAME/welding-defect-yolov8-full-training")
results = model("image.jpg")
results[0].show()

avinashhm/welding-defect-yolov8-full-training

作者 avinashhm

object-detection ultralytics
↓ 9 ♥ 0

创建时间: 2025-06-02 07:57:11+00:00

更新时间: 2025-06-02 08:08:08+00:00

在 Hugging Face 上查看

文件 (28)

.gitattributes
F1_curve.png
PR_curve.png
P_curve.png
README.md
R_curve.png
args.yaml
confusion_matrix.png
confusion_matrix_normalized.png
labels.jpg
labels_correlogram.jpg
results.csv
results.png
train_batch0.jpg
train_batch1.jpg
train_batch2.jpg
train_batch9180.jpg
train_batch9181.jpg
train_batch9182.jpg
val_batch0_labels.jpg
val_batch0_pred.jpg
val_batch1_labels.jpg
val_batch1_pred.jpg
val_batch2_labels.jpg
val_batch2_pred.jpg
weights/best.onnx ONNX
weights/best.pt
weights/last.pt