返回模型
说明文档
焊接缺陷检测模型 (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