Upload 24 files
Browse files- yolov5x_custom/runs/detect/yolov5x_custom/F1_curve.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/PR_curve.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/P_curve.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/R_curve.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/args.yaml +106 -0
- yolov5x_custom/runs/detect/yolov5x_custom/confusion_matrix.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/confusion_matrix_normalized.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/events.out.tfevents.1716627839.1cdfa80caa1e.6433.0 +3 -0
- yolov5x_custom/runs/detect/yolov5x_custom/labels.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/labels_correlogram.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/results.csv +2 -0
- yolov5x_custom/runs/detect/yolov5x_custom/results.png +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/train5xTensorboard.py +48 -0
- yolov5x_custom/runs/detect/yolov5x_custom/train_batch0.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/train_batch1.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/train_batch2.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/val_batch0_labels.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/val_batch0_pred.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/val_batch1_labels.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/val_batch1_pred.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/val_batch2_labels.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/val_batch2_pred.jpg +0 -0
- yolov5x_custom/runs/detect/yolov5x_custom/weights/best.pt +3 -0
- yolov5x_custom/runs/detect/yolov5x_custom/weights/last.pt +3 -0
yolov5x_custom/runs/detect/yolov5x_custom/F1_curve.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/PR_curve.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/P_curve.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/R_curve.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/args.yaml
ADDED
@@ -0,0 +1,106 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
task: detect
|
2 |
+
mode: train
|
3 |
+
model: yolov5x.pt
|
4 |
+
data: data/data.yaml
|
5 |
+
epochs: 1
|
6 |
+
time: null
|
7 |
+
patience: 100
|
8 |
+
batch: 4
|
9 |
+
imgsz: 1024
|
10 |
+
save: true
|
11 |
+
save_period: -1
|
12 |
+
cache: false
|
13 |
+
device: null
|
14 |
+
workers: 8
|
15 |
+
project: null
|
16 |
+
name: yolov5x_custom
|
17 |
+
exist_ok: false
|
18 |
+
pretrained: true
|
19 |
+
optimizer: auto
|
20 |
+
verbose: true
|
21 |
+
seed: 0
|
22 |
+
deterministic: true
|
23 |
+
single_cls: false
|
24 |
+
rect: false
|
25 |
+
cos_lr: false
|
26 |
+
close_mosaic: 10
|
27 |
+
resume: false
|
28 |
+
amp: true
|
29 |
+
fraction: 1.0
|
30 |
+
profile: false
|
31 |
+
freeze: null
|
32 |
+
multi_scale: false
|
33 |
+
overlap_mask: true
|
34 |
+
mask_ratio: 4
|
35 |
+
dropout: 0.0
|
36 |
+
val: true
|
37 |
+
split: val
|
38 |
+
save_json: false
|
39 |
+
save_hybrid: false
|
40 |
+
conf: null
|
41 |
+
iou: 0.2
|
42 |
+
max_det: 300
|
43 |
+
half: false
|
44 |
+
dnn: false
|
45 |
+
plots: true
|
46 |
+
source: null
|
47 |
+
vid_stride: 1
|
48 |
+
stream_buffer: false
|
49 |
+
visualize: false
|
50 |
+
augment: false
|
51 |
+
agnostic_nms: false
|
52 |
+
classes: null
|
53 |
+
retina_masks: false
|
54 |
+
embed: null
|
55 |
+
show: false
|
56 |
+
save_frames: false
|
57 |
+
save_txt: false
|
58 |
+
save_conf: false
|
59 |
+
save_crop: false
|
60 |
+
show_labels: true
|
61 |
+
show_conf: true
|
62 |
+
show_boxes: true
|
63 |
+
line_width: null
|
64 |
+
format: torchscript
|
65 |
+
keras: false
|
66 |
+
optimize: false
|
67 |
+
int8: false
|
68 |
+
dynamic: false
|
69 |
+
simplify: false
|
70 |
+
opset: null
|
71 |
+
workspace: 4
|
72 |
+
nms: false
|
73 |
+
lr0: 0.01
|
74 |
+
lrf: 0.01
|
75 |
+
momentum: 0.937
|
76 |
+
weight_decay: 0.0005
|
77 |
+
warmup_epochs: 1.0
|
78 |
+
warmup_momentum: 0.8
|
79 |
+
warmup_bias_lr: 0.1
|
80 |
+
box: 0.02
|
81 |
+
cls: 0.5
|
82 |
+
dfl: 1.5
|
83 |
+
pose: 12.0
|
84 |
+
kobj: 1.0
|
85 |
+
label_smoothing: 0.0
|
86 |
+
nbs: 64
|
87 |
+
hsv_h: 0.015
|
88 |
+
hsv_s: 0.7
|
89 |
+
hsv_v: 0.4
|
90 |
+
degrees: 0.0
|
91 |
+
translate: 0.1
|
92 |
+
scale: 0.5
|
93 |
+
shear: 0.0
|
94 |
+
perspective: 0.0
|
95 |
+
flipud: 0.0
|
96 |
+
fliplr: 0.5
|
97 |
+
bgr: 0.0
|
98 |
+
mosaic: 0.5
|
99 |
+
mixup: 0.0
|
100 |
+
copy_paste: 0.0
|
101 |
+
auto_augment: randaugment
|
102 |
+
erasing: 0.4
|
103 |
+
crop_fraction: 1.0
|
104 |
+
cfg: null
|
105 |
+
tracker: botsort.yaml
|
106 |
+
save_dir: runs/detect/yolov5x_custom
|
yolov5x_custom/runs/detect/yolov5x_custom/confusion_matrix.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/confusion_matrix_normalized.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/events.out.tfevents.1716627839.1cdfa80caa1e.6433.0
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e87f7e1431ee17e5a3d3e5c940d3a9f69f478642e0d3bd2801aaaaf340d77f1
|
3 |
+
size 396915
|
yolov5x_custom/runs/detect/yolov5x_custom/labels.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/labels_correlogram.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/results.csv
ADDED
@@ -0,0 +1,2 @@
|
|
|
|
|
|
|
1 |
+
epoch, train/box_loss, train/cls_loss, train/dfl_loss, metrics/precision(B), metrics/recall(B), metrics/mAP50(B), metrics/mAP50-95(B), val/box_loss, val/cls_loss, val/dfl_loss, lr/pg0, lr/pg1, lr/pg2
|
2 |
+
1, 0.00324, 2.8828, 1.5265, 0.32347, 0.17336, 0.14169, 0.08413, 0.00407, 3.7808, 1.9341, 0.00018193, 0.00018193, 0.00018193
|
yolov5x_custom/runs/detect/yolov5x_custom/results.png
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/train5xTensorboard.py
ADDED
@@ -0,0 +1,48 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import torch
|
2 |
+
from tensorboardX import SummaryWriter
|
3 |
+
from ultralytics import YOLO
|
4 |
+
|
5 |
+
# TensorBoardX SummaryWriter ์ค์
|
6 |
+
writer = SummaryWriter('runs/yolov8_example')
|
7 |
+
|
8 |
+
# ์ฌ์ ํ์ต๋ YOLOv5 ๋ชจ๋ธ ๋ก๋
|
9 |
+
model = YOLO('yolov8n.pt')
|
10 |
+
|
11 |
+
# ํ์ต ์ค์
|
12 |
+
epochs = 10
|
13 |
+
data_path = 'data/data.yaml'
|
14 |
+
img_size = 640
|
15 |
+
batch_size = 4
|
16 |
+
project_name = 'yolov8n_custom'
|
17 |
+
project_dir = 'runs/train'
|
18 |
+
|
19 |
+
# ์ฌ์ฉ์ ์ ์ ํ์ต ํจ์
|
20 |
+
def train_yolov5(model, data_path, img_size, batch_size, epochs, project_name, project_dir):
|
21 |
+
results = model.train(
|
22 |
+
data=data_path,
|
23 |
+
imgsz=img_size,
|
24 |
+
epochs=epochs,
|
25 |
+
batch=batch_size,
|
26 |
+
name=project_name,
|
27 |
+
project=project_dir
|
28 |
+
)
|
29 |
+
metrics = results
|
30 |
+
writer.add_scalar('Loss/box', metrics.box_loss, 1)
|
31 |
+
writer.add_scalar('Loss/obj', metrics.obj_loss, 1)
|
32 |
+
writer.add_scalar('Loss/cls', metrics.cls_loss, 1)
|
33 |
+
writer.add_scalar('Loss/total', metrics.total_loss, 1)
|
34 |
+
|
35 |
+
writer.add_scalar('Precision', metrics.precision, 1)
|
36 |
+
writer.add_scalar('Recall', metrics.recall, 1)
|
37 |
+
writer.add_scalar('mAP@0.5', metrics.map50, 1)
|
38 |
+
writer.add_scalar('mAP@0.5:0.95', metrics.map, 1)
|
39 |
+
|
40 |
+
# TensorBoardX writer ๋ซ๊ธฐ
|
41 |
+
writer.close()
|
42 |
+
|
43 |
+
# ๋ชจ๋ธ ํ์ต
|
44 |
+
train_yolov5(model, data_path, img_size, batch_size, epochs, project_name, project_dir)
|
45 |
+
|
46 |
+
# TensorBoard ์คํ
|
47 |
+
%load_ext tensorboard
|
48 |
+
%tensorboard --logdir=runs/yolov8_example
|
yolov5x_custom/runs/detect/yolov5x_custom/train_batch0.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/train_batch1.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/train_batch2.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/val_batch0_labels.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/val_batch0_pred.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/val_batch1_labels.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/val_batch1_pred.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/val_batch2_labels.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/val_batch2_pred.jpg
ADDED
yolov5x_custom/runs/detect/yolov5x_custom/weights/best.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:9e626d25ad4480f5a7c8009476289f15de2253eb51acc0d87ea4632d5f9aa92d
|
3 |
+
size 195083125
|
yolov5x_custom/runs/detect/yolov5x_custom/weights/last.pt
ADDED
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
1 |
+
version https://git-lfs.github.com/spec/v1
|
2 |
+
oid sha256:ef86ac4a8097d8a2a4c375bc621078f34772dfef7849ce493f05d6dac8638646
|
3 |
+
size 195083125
|