avinashhm/welding-defect-yolov8

Welding Defect Detection Model (YOLOv8)

This is a fine-tuned YOLOv8 model for detecting welding defects such as:

• good weld
• bad weld
• defect

Trained using Ultralytics YOLO on a custom dataset of welding images.

Training Details

• Model Type: YOLOv8
• Dataset: The Welding Defect Dataset - v2
• Classes: 3
• Epochs: 100
• Image Size: 640x640
• Batch Size: 16

Usage

from ultralytics import YOLO

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