Alex-Watchman commited on
Commit
bdefd1f
·
verified ·
1 Parent(s): ea7789a

Added weights file, configuration file, and a demo README.md

Browse files
Files changed (3) hide show
  1. README.md +75 -0
  2. basic_config.yaml +42 -0
  3. 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