Added weights file, configuration file, and a demo README.md
Browse files- README.md +75 -0
- basic_config.yaml +42 -0
- best.pt +3 -0
README.md
CHANGED
|
@@ -1,3 +1,78 @@
|
|
| 1 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 2 |
license: mit
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 3 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
+
language:
|
| 3 |
+
- en
|
| 4 |
+
- ru
|
| 5 |
+
tags:
|
| 6 |
+
- computer-vision
|
| 7 |
+
- object-detection
|
| 8 |
+
- yolov8
|
| 9 |
+
- welding
|
| 10 |
+
- ndt
|
| 11 |
+
- defect-detection
|
| 12 |
license: mit
|
| 13 |
+
datasets:
|
| 14 |
+
- synthetic-welding-defects
|
| 15 |
+
metrics:
|
| 16 |
+
- precision
|
| 17 |
+
- recall
|
| 18 |
+
- mAP
|
| 19 |
---
|
| 20 |
+
|
| 21 |
+
# XVL: X-Ray Vision Lab - Welding Defect Detector
|
| 22 |
+
|
| 23 |
+
YOLOv8-based model for automated detection of welding defects in X-ray images.
|
| 24 |
+
|
| 25 |
+
## Model Details
|
| 26 |
+
- **Architecture**: YOLOv8n (custom)
|
| 27 |
+
- **Input Size**: 512x512
|
| 28 |
+
- **Classes**: 5 defect types
|
| 29 |
+
- **Training Data**: Synthetic X-ray images (8000+ samples)
|
| 30 |
+
- **Validation Data**: Real industrial X-ray scans (200+)
|
| 31 |
+
|
| 32 |
+
## Performance
|
| 33 |
+
| Metric | Value | Epoch |
|
| 34 |
+
|------------|--------|-------|
|
| 35 |
+
| Precision | 95.6% | 37 |
|
| 36 |
+
| Recall | 88.9% | 39 |
|
| 37 |
+
| mAP@50 | 93.3% | 39 |
|
| 38 |
+
| mAP@50-95 | 78.8% | 37 |
|
| 39 |
+
|
| 40 |
+
## Usage
|
| 41 |
+
|
| 42 |
+
### With PyTorch
|
| 43 |
+
```python
|
| 44 |
+
import torch
|
| 45 |
+
from models.yolo_custom import load_model
|
| 46 |
+
|
| 47 |
+
model = load_model(
|
| 48 |
+
weights="best.pt",
|
| 49 |
+
config="config.yaml"
|
| 50 |
+
)
|
| 51 |
+
With XVL Project
|
| 52 |
+
bash
|
| 53 |
+
git clone https://github.com/your-username/XVL.git
|
| 54 |
+
python scripts/download_weights.py
|
| 55 |
+
python src/run.py
|
| 56 |
+
Defect Classes
|
| 57 |
+
Cracks
|
| 58 |
+
|
| 59 |
+
Porosity
|
| 60 |
+
|
| 61 |
+
Lack of penetration
|
| 62 |
+
|
| 63 |
+
Slag inclusions
|
| 64 |
+
|
| 65 |
+
Undercut
|
| 66 |
+
|
| 67 |
+
Training Configuration
|
| 68 |
+
See config.yaml for full details.
|
| 69 |
+
|
| 70 |
+
Citation
|
| 71 |
+
If you use this model, please reference:
|
| 72 |
+
|
| 73 |
+
@software{xvl2024,
|
| 74 |
+
title={XVL: X-Ray Vision Lab},
|
| 75 |
+
author={Your Name},
|
| 76 |
+
year={2024},
|
| 77 |
+
url={https://github.com/your-username/XVL}
|
| 78 |
+
}
|
basic_config.yaml
ADDED
|
@@ -0,0 +1,42 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# configs.yaml
|
| 2 |
+
training:
|
| 3 |
+
img_size: 512
|
| 4 |
+
model_name: "yolov8s.pt"
|
| 5 |
+
epochs: 150
|
| 6 |
+
batch_size: 48
|
| 7 |
+
learning_rate: 0.00005
|
| 8 |
+
num_folds: 5
|
| 9 |
+
|
| 10 |
+
data:
|
| 11 |
+
samples_per_epoch: null
|
| 12 |
+
val_samples: null
|
| 13 |
+
class_weights: null
|
| 14 |
+
class_names:
|
| 15 |
+
- "incomplete_fusion"
|
| 16 |
+
- "crack"
|
| 17 |
+
- "single_pore"
|
| 18 |
+
- "cluster_pores"
|
| 19 |
+
- "empty"
|
| 20 |
+
|
| 21 |
+
# Параметры обучения YOLO
|
| 22 |
+
yolo_args:
|
| 23 |
+
save: true
|
| 24 |
+
save_period: 10
|
| 25 |
+
exist_ok: true
|
| 26 |
+
pretrained: true
|
| 27 |
+
optimizer: "AdamW"
|
| 28 |
+
weight_decay: 0.005
|
| 29 |
+
dropout: 0.1
|
| 30 |
+
mixup: 0.05
|
| 31 |
+
cutmix: 0.05
|
| 32 |
+
warmup_epochs: 20
|
| 33 |
+
cos_lr: true
|
| 34 |
+
label_smoothing: 0.1
|
| 35 |
+
patience: 30
|
| 36 |
+
verbose: true
|
| 37 |
+
# Другие параметры YOLO могут быть добавлены здесь
|
| 38 |
+
|
| 39 |
+
paths:
|
| 40 |
+
project_dir: "models"
|
| 41 |
+
log_dir: "logs/training"
|
| 42 |
+
metrics_dir: "metrics"
|
best.pt
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:72f3c76d2fe2c4fe6fbfa65af9de65da528bef33a46967e4f20b0c1cc78ccc84
|
| 3 |
+
size 51997906
|