ignore datasets
@ -1,4 +1,7 @@
|
|||||||
<?xml version="1.0" encoding="UTF-8"?>
|
<?xml version="1.0" encoding="UTF-8"?>
|
||||||
<project version="4">
|
<project version="4">
|
||||||
|
<component name="Black">
|
||||||
|
<option name="sdkName" value="Python 3.10.12 WSL (Ubuntu-22.04): (/home/awin/python/venvs/model/bin/python)" />
|
||||||
|
</component>
|
||||||
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13 (models)" project-jdk-type="Python SDK" />
|
<component name="ProjectRootManager" version="2" project-jdk-name="Python 3.13 (models)" project-jdk-type="Python SDK" />
|
||||||
</project>
|
</project>
|
||||||
106
yolov11/runs/detect/train/args.yaml
Normal file
@ -0,0 +1,106 @@
|
|||||||
|
task: detect
|
||||||
|
mode: train
|
||||||
|
model: yolo12n.pt
|
||||||
|
data: coco8.yaml
|
||||||
|
epochs: 10
|
||||||
|
time: null
|
||||||
|
patience: 100
|
||||||
|
batch: 16
|
||||||
|
imgsz: 640
|
||||||
|
save: true
|
||||||
|
save_period: -1
|
||||||
|
cache: false
|
||||||
|
device: null
|
||||||
|
workers: 8
|
||||||
|
project: null
|
||||||
|
name: train
|
||||||
|
exist_ok: false
|
||||||
|
pretrained: true
|
||||||
|
optimizer: auto
|
||||||
|
verbose: true
|
||||||
|
seed: 0
|
||||||
|
deterministic: true
|
||||||
|
single_cls: false
|
||||||
|
rect: false
|
||||||
|
cos_lr: false
|
||||||
|
close_mosaic: 10
|
||||||
|
resume: false
|
||||||
|
amp: true
|
||||||
|
fraction: 1.0
|
||||||
|
profile: false
|
||||||
|
freeze: null
|
||||||
|
multi_scale: false
|
||||||
|
overlap_mask: true
|
||||||
|
mask_ratio: 4
|
||||||
|
dropout: 0.0
|
||||||
|
val: true
|
||||||
|
split: val
|
||||||
|
save_json: false
|
||||||
|
save_hybrid: false
|
||||||
|
conf: null
|
||||||
|
iou: 0.7
|
||||||
|
max_det: 300
|
||||||
|
half: false
|
||||||
|
dnn: false
|
||||||
|
plots: true
|
||||||
|
source: null
|
||||||
|
vid_stride: 1
|
||||||
|
stream_buffer: false
|
||||||
|
visualize: false
|
||||||
|
augment: false
|
||||||
|
agnostic_nms: false
|
||||||
|
classes: null
|
||||||
|
retina_masks: false
|
||||||
|
embed: null
|
||||||
|
show: false
|
||||||
|
save_frames: false
|
||||||
|
save_txt: false
|
||||||
|
save_conf: false
|
||||||
|
save_crop: false
|
||||||
|
show_labels: true
|
||||||
|
show_conf: true
|
||||||
|
show_boxes: true
|
||||||
|
line_width: null
|
||||||
|
format: torchscript
|
||||||
|
keras: false
|
||||||
|
optimize: false
|
||||||
|
int8: false
|
||||||
|
dynamic: false
|
||||||
|
simplify: true
|
||||||
|
opset: null
|
||||||
|
workspace: null
|
||||||
|
nms: false
|
||||||
|
lr0: 0.01
|
||||||
|
lrf: 0.01
|
||||||
|
momentum: 0.937
|
||||||
|
weight_decay: 0.0005
|
||||||
|
warmup_epochs: 3.0
|
||||||
|
warmup_momentum: 0.8
|
||||||
|
warmup_bias_lr: 0.1
|
||||||
|
box: 7.5
|
||||||
|
cls: 0.5
|
||||||
|
dfl: 1.5
|
||||||
|
pose: 12.0
|
||||||
|
kobj: 1.0
|
||||||
|
nbs: 64
|
||||||
|
hsv_h: 0.015
|
||||||
|
hsv_s: 0.7
|
||||||
|
hsv_v: 0.4
|
||||||
|
degrees: 0.0
|
||||||
|
translate: 0.1
|
||||||
|
scale: 0.5
|
||||||
|
shear: 0.0
|
||||||
|
perspective: 0.0
|
||||||
|
flipud: 0.0
|
||||||
|
fliplr: 0.5
|
||||||
|
bgr: 0.0
|
||||||
|
mosaic: 1.0
|
||||||
|
mixup: 0.0
|
||||||
|
copy_paste: 0.0
|
||||||
|
copy_paste_mode: flip
|
||||||
|
auto_augment: randaugment
|
||||||
|
erasing: 0.4
|
||||||
|
crop_fraction: 1.0
|
||||||
|
cfg: null
|
||||||
|
tracker: botsort.yaml
|
||||||
|
save_dir: runs/detect/train
|
||||||
BIN
yolov11/runs/detect/train2/F1_curve.png
Normal file
|
After Width: | Height: | Size: 157 KiB |
BIN
yolov11/runs/detect/train2/PR_curve.png
Normal file
|
After Width: | Height: | Size: 107 KiB |
BIN
yolov11/runs/detect/train2/P_curve.png
Normal file
|
After Width: | Height: | Size: 178 KiB |
BIN
yolov11/runs/detect/train2/R_curve.png
Normal file
|
After Width: | Height: | Size: 134 KiB |
106
yolov11/runs/detect/train2/args.yaml
Normal file
@ -0,0 +1,106 @@
|
|||||||
|
task: detect
|
||||||
|
mode: train
|
||||||
|
model: yolo12n.pt
|
||||||
|
data: coco8.yaml
|
||||||
|
epochs: 10
|
||||||
|
time: null
|
||||||
|
patience: 100
|
||||||
|
batch: 16
|
||||||
|
imgsz: 640
|
||||||
|
save: true
|
||||||
|
save_period: -1
|
||||||
|
cache: false
|
||||||
|
device: null
|
||||||
|
workers: 8
|
||||||
|
project: null
|
||||||
|
name: train2
|
||||||
|
exist_ok: false
|
||||||
|
pretrained: true
|
||||||
|
optimizer: auto
|
||||||
|
verbose: true
|
||||||
|
seed: 0
|
||||||
|
deterministic: true
|
||||||
|
single_cls: false
|
||||||
|
rect: false
|
||||||
|
cos_lr: false
|
||||||
|
close_mosaic: 10
|
||||||
|
resume: false
|
||||||
|
amp: true
|
||||||
|
fraction: 1.0
|
||||||
|
profile: false
|
||||||
|
freeze: null
|
||||||
|
multi_scale: false
|
||||||
|
overlap_mask: true
|
||||||
|
mask_ratio: 4
|
||||||
|
dropout: 0.0
|
||||||
|
val: true
|
||||||
|
split: val
|
||||||
|
save_json: false
|
||||||
|
save_hybrid: false
|
||||||
|
conf: null
|
||||||
|
iou: 0.7
|
||||||
|
max_det: 300
|
||||||
|
half: false
|
||||||
|
dnn: false
|
||||||
|
plots: true
|
||||||
|
source: null
|
||||||
|
vid_stride: 1
|
||||||
|
stream_buffer: false
|
||||||
|
visualize: false
|
||||||
|
augment: false
|
||||||
|
agnostic_nms: false
|
||||||
|
classes: null
|
||||||
|
retina_masks: false
|
||||||
|
embed: null
|
||||||
|
show: false
|
||||||
|
save_frames: false
|
||||||
|
save_txt: false
|
||||||
|
save_conf: false
|
||||||
|
save_crop: false
|
||||||
|
show_labels: true
|
||||||
|
show_conf: true
|
||||||
|
show_boxes: true
|
||||||
|
line_width: null
|
||||||
|
format: torchscript
|
||||||
|
keras: false
|
||||||
|
optimize: false
|
||||||
|
int8: false
|
||||||
|
dynamic: false
|
||||||
|
simplify: true
|
||||||
|
opset: null
|
||||||
|
workspace: null
|
||||||
|
nms: false
|
||||||
|
lr0: 0.01
|
||||||
|
lrf: 0.01
|
||||||
|
momentum: 0.937
|
||||||
|
weight_decay: 0.0005
|
||||||
|
warmup_epochs: 3.0
|
||||||
|
warmup_momentum: 0.8
|
||||||
|
warmup_bias_lr: 0.1
|
||||||
|
box: 7.5
|
||||||
|
cls: 0.5
|
||||||
|
dfl: 1.5
|
||||||
|
pose: 12.0
|
||||||
|
kobj: 1.0
|
||||||
|
nbs: 64
|
||||||
|
hsv_h: 0.015
|
||||||
|
hsv_s: 0.7
|
||||||
|
hsv_v: 0.4
|
||||||
|
degrees: 0.0
|
||||||
|
translate: 0.1
|
||||||
|
scale: 0.5
|
||||||
|
shear: 0.0
|
||||||
|
perspective: 0.0
|
||||||
|
flipud: 0.0
|
||||||
|
fliplr: 0.5
|
||||||
|
bgr: 0.0
|
||||||
|
mosaic: 1.0
|
||||||
|
mixup: 0.0
|
||||||
|
copy_paste: 0.0
|
||||||
|
copy_paste_mode: flip
|
||||||
|
auto_augment: randaugment
|
||||||
|
erasing: 0.4
|
||||||
|
crop_fraction: 1.0
|
||||||
|
cfg: null
|
||||||
|
tracker: botsort.yaml
|
||||||
|
save_dir: runs/detect/train2
|
||||||
BIN
yolov11/runs/detect/train2/confusion_matrix.png
Normal file
|
After Width: | Height: | Size: 536 KiB |
BIN
yolov11/runs/detect/train2/confusion_matrix_normalized.png
Normal file
|
After Width: | Height: | Size: 539 KiB |
BIN
yolov11/runs/detect/train2/labels.jpg
Normal file
|
After Width: | Height: | Size: 97 KiB |
BIN
yolov11/runs/detect/train2/labels_correlogram.jpg
Normal file
|
After Width: | Height: | Size: 179 KiB |
11
yolov11/runs/detect/train2/results.csv
Normal file
@ -0,0 +1,11 @@
|
|||||||
|
epoch,time,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
|
||||||
|
1,6.86948,0.70939,1.05707,1.14039,0.02628,0.21667,0.08764,0.05376,2.11971,4.98083,2.04794,0,0,0
|
||||||
|
2,11.0357,0.67899,1.18625,1.09058,0.05949,0.3,0.12129,0.05636,2.05773,4.70166,2.00079,1.07219e-06,1.07219e-06,1.07219e-06
|
||||||
|
3,14.7562,0.90865,1.26633,1.20842,0.06927,0.56667,0.13907,0.06017,1.97415,4.37446,1.95573,1.90876e-06,1.90876e-06,1.90876e-06
|
||||||
|
4,18.1024,0.74725,0.99608,1.08456,0.06112,0.56667,0.24832,0.08509,1.9214,4.18075,1.89617,2.50971e-06,2.50971e-06,2.50971e-06
|
||||||
|
5,21.4479,0.70276,1.00317,1.09192,0.14125,0.36109,0.2597,0.18404,1.90686,3.91017,1.86089,2.87504e-06,2.87504e-06,2.87504e-06
|
||||||
|
6,24.8385,0.79145,0.81997,1.07367,0.54119,0.24116,0.26546,0.18547,1.92964,3.65661,1.84599,3.00475e-06,3.00475e-06,3.00475e-06
|
||||||
|
7,28.4703,0.89764,1.04108,1.21462,0.52741,0.32658,0.27542,0.18197,1.83827,3.44983,1.72672,2.89884e-06,2.89884e-06,2.89884e-06
|
||||||
|
8,31.9447,0.94011,0.94038,1.21287,0.54095,0.34005,0.29672,0.20452,1.78655,3.25106,1.70385,2.55731e-06,2.55731e-06,2.55731e-06
|
||||||
|
9,35.7747,0.83725,0.83766,1.22182,0.55701,0.35369,0.31601,0.22002,1.69535,3.09446,1.59676,1.98016e-06,1.98016e-06,1.98016e-06
|
||||||
|
10,40.787,0.81827,0.87845,1.09094,0.56054,0.36667,0.32583,0.24564,1.63002,2.96068,1.54419,1.16739e-06,1.16739e-06,1.16739e-06
|
||||||
|
BIN
yolov11/runs/detect/train2/results.png
Normal file
|
After Width: | Height: | Size: 284 KiB |
BIN
yolov11/runs/detect/train2/train_batch0.jpg
Normal file
|
After Width: | Height: | Size: 315 KiB |
BIN
yolov11/runs/detect/train2/train_batch1.jpg
Normal file
|
After Width: | Height: | Size: 311 KiB |
BIN
yolov11/runs/detect/train2/train_batch2.jpg
Normal file
|
After Width: | Height: | Size: 230 KiB |
BIN
yolov11/runs/detect/train2/val_batch0_labels.jpg
Normal file
|
After Width: | Height: | Size: 377 KiB |
BIN
yolov11/runs/detect/train2/val_batch0_pred.jpg
Normal file
|
After Width: | Height: | Size: 373 KiB |
BIN
yolov11/runs/detect/train2/weights/best.onnx
Normal file
BIN
yolov11/runs/detect/train2/weights/best.pt
Normal file
BIN
yolov11/runs/detect/train2/weights/last.pt
Normal file
BIN
yolov11/runs/detect/train22/F1_curve.png
Normal file
|
After Width: | Height: | Size: 157 KiB |
BIN
yolov11/runs/detect/train22/PR_curve.png
Normal file
|
After Width: | Height: | Size: 107 KiB |
BIN
yolov11/runs/detect/train22/P_curve.png
Normal file
|
After Width: | Height: | Size: 178 KiB |
BIN
yolov11/runs/detect/train22/R_curve.png
Normal file
|
After Width: | Height: | Size: 134 KiB |
BIN
yolov11/runs/detect/train22/confusion_matrix.png
Normal file
|
After Width: | Height: | Size: 536 KiB |
BIN
yolov11/runs/detect/train22/confusion_matrix_normalized.png
Normal file
|
After Width: | Height: | Size: 539 KiB |
BIN
yolov11/runs/detect/train22/val_batch0_labels.jpg
Normal file
|
After Width: | Height: | Size: 377 KiB |
BIN
yolov11/runs/detect/train22/val_batch0_pred.jpg
Normal file
|
After Width: | Height: | Size: 373 KiB |
@ -4,7 +4,7 @@ from ultralytics import YOLO
|
|||||||
model = YOLO("yolo12n.pt")
|
model = YOLO("yolo12n.pt")
|
||||||
|
|
||||||
# Train the model on the COCO8 example dataset for 100 epochs
|
# Train the model on the COCO8 example dataset for 100 epochs
|
||||||
results = model.train(data="coco8.yaml", epochs=100, imgsz=640)
|
results = model.train(data="coco8.yaml", epochs=10, imgsz=640)
|
||||||
print(results)
|
print(results)
|
||||||
|
|
||||||
# Evaluate the model's performance on the validation set
|
# Evaluate the model's performance on the validation set
|
||||||
|
|||||||