| library_name: ultralytics | |
| tags: | |
| - object-detection | |
| - yolov8 | |
| - welding-defect | |
| - industrial-inspection | |
| # 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 | |
| ```python | |
| from ultralytics import YOLO | |
| model = YOLO("YOUR_HF_USERNAME/welding-defect-yolov8") | |
| results = model("image.jpg") | |
| results[0].show() | |