diff --git a/best.pt b/best.pt new file mode 100644 index 0000000..402ccba Binary files /dev/null and b/best.pt differ diff --git a/train_temp/BoxF1_curve.png b/train_temp/BoxF1_curve.png new file mode 100644 index 0000000..b1c0c3b Binary files /dev/null and b/train_temp/BoxF1_curve.png differ diff --git a/train_temp/BoxPR_curve.png b/train_temp/BoxPR_curve.png new file mode 100644 index 0000000..4ed30b9 Binary files /dev/null and b/train_temp/BoxPR_curve.png differ diff --git a/train_temp/BoxP_curve.png b/train_temp/BoxP_curve.png new file mode 100644 index 0000000..ef2c3a4 Binary files /dev/null and b/train_temp/BoxP_curve.png differ diff --git a/train_temp/BoxR_curve.png b/train_temp/BoxR_curve.png new file mode 100644 index 0000000..51a2324 Binary files /dev/null and b/train_temp/BoxR_curve.png differ diff --git a/train_temp/args.yaml b/train_temp/args.yaml new file mode 100644 index 0000000..31d50f1 --- /dev/null +++ b/train_temp/args.yaml @@ -0,0 +1,109 @@ +task: detect +mode: train +model: /home/jovyan/model/leejaewoo/yolo11/yolo11m.pt +data: data.yaml +epochs: 20 +time: null +patience: 100 +batch: 16 +imgsz: 640 +save: true +save_period: -1 +cache: false +device: null +workers: 8 +project: /home/jovyan/model/leejaewoo/yolo11 +name: train_temp +exist_ok: true +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: 0.0 +compile: false +overlap_mask: true +mask_ratio: 4 +dropout: 0.0 +val: true +split: val +save_json: false +conf: null +iou: 0.7 +max_det: 300 +half: false +dnn: false +plots: true +end2end: null +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 +rle: 1.0 +angle: 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 +cutmix: 0.0 +copy_paste: 0.0 +copy_paste_mode: flip +auto_augment: randaugment +erasing: 0.4 +cfg: null +tracker: botsort.yaml +save_dir: /home/jovyan/model/leejaewoo/yolo11/train_temp diff --git a/train_temp/confusion_matrix.png b/train_temp/confusion_matrix.png new file mode 100644 index 0000000..5098985 Binary files /dev/null and b/train_temp/confusion_matrix.png differ diff --git a/train_temp/confusion_matrix_normalized.png b/train_temp/confusion_matrix_normalized.png new file mode 100644 index 0000000..68117d6 Binary files /dev/null and b/train_temp/confusion_matrix_normalized.png differ diff --git a/train_temp/labels.jpg b/train_temp/labels.jpg new file mode 100644 index 0000000..2fdb6d7 Binary files /dev/null and b/train_temp/labels.jpg differ diff --git a/train_temp/results.csv b/train_temp/results.csv new file mode 100644 index 0000000..df0680c --- /dev/null +++ b/train_temp/results.csv @@ -0,0 +1,21 @@ +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,1.67206,1.20329,2.63505,1.60228,0.00426,0.33333,0.33801,0.23587,1.68014,2.95409,1.18407,0,0,0 +2,2.49076,1.71571,3.1574,2.06722,0.00432,0.33333,0.33804,0.23588,1.68943,2.93518,1.18411,1.901e-05,1.901e-05,1.901e-05 +3,3.44053,1.28664,2.76427,1.80032,0.00394,0.33333,0.338,0.23586,1.68377,2.92535,1.18199,3.604e-05,3.604e-05,3.604e-05 +4,4.22345,2.02959,2.78247,2.12608,0.00384,0.33333,0.33802,0.23588,1.68336,2.91587,1.18185,5.109e-05,5.109e-05,5.109e-05 +5,5.11071,1.29589,2.4997,1.52708,0.00378,0.33333,0.33801,0.23587,1.62366,2.8887,1.17599,6.416e-05,6.416e-05,6.416e-05 +6,5.89156,1.51277,3.06007,1.79636,0.00367,0.33333,0.3393,0.27007,1.62356,2.88391,1.17534,7.525e-05,7.525e-05,7.525e-05 +7,6.78513,1.16295,2.88673,1.43502,0.00701,0.66667,0.34071,0.27069,1.61951,2.85114,1.17647,8.436e-05,8.436e-05,8.436e-05 +8,7.69055,1.80226,3.3897,2.19002,0.005,1,0.34149,0.27104,1.61407,2.82281,1.17371,9.149e-05,9.149e-05,9.149e-05 +9,8.66732,1.36534,2.83636,1.59469,1,0.92064,0.995,0.66472,1.44301,1.35205,1.08568,9.664e-05,9.664e-05,9.664e-05 +10,10.4185,0.83717,1.591,1.14281,0.97649,1,0.995,0.73056,0.99007,1.32467,0.99136,9.981e-05,9.981e-05,9.981e-05 +11,13.2569,1.13175,2.83905,1.40507,0.96256,1,0.995,0.50029,1.23615,1.68106,1.01363,0.000101,0.000101,0.000101 +12,14.6748,0.61373,2.15461,1.21324,0.96159,1,0.995,0.69789,1.21433,1.15714,0.97342,0.00010021,0.00010021,0.00010021 +13,16.0372,1.13271,2.03106,1.14791,0.9764,0.66667,0.83,0.47642,1.27881,1.4907,1.08621,9.744e-05,9.744e-05,9.744e-05 +14,17.3915,1.0798,2.2862,1.60101,0.74727,0.98909,0.83,0.61522,1.12672,1.55495,1.0186,9.269e-05,9.269e-05,9.269e-05 +15,18.7863,0.69968,1.19794,1.06678,0.66158,0.66667,0.56147,0.50237,1.08498,1.72119,1.02719,8.596e-05,8.596e-05,8.596e-05 +16,20.142,0.58534,1.12073,0.99467,0.66516,0.66366,0.44389,0.32753,1.57405,2.30618,1.12398,7.725e-05,7.725e-05,7.725e-05 +17,21.4689,0.62205,0.87617,0.97014,0.66389,0.66112,0.55556,0.36689,1.89155,2.11945,1.19579,6.656e-05,6.656e-05,6.656e-05 +18,22.8171,0.53723,0.9658,0.69231,0.66389,0.66112,0.55556,0.36689,1.89155,2.11945,1.19579,5.389e-05,5.389e-05,5.389e-05 +19,24.237,0.70819,0.95772,1.17249,0.66353,0.66039,0.56738,0.36925,1.6714,1.91917,1.13025,3.924e-05,3.924e-05,3.924e-05 +20,25.5601,0.67334,0.98455,0.87524,0.66353,0.66039,0.56738,0.36925,1.6714,1.91917,1.13025,2.261e-05,2.261e-05,2.261e-05 diff --git a/train_temp/results.png b/train_temp/results.png new file mode 100644 index 0000000..70ae0d3 Binary files /dev/null and b/train_temp/results.png differ diff --git a/train_temp/train_batch0.jpg b/train_temp/train_batch0.jpg new file mode 100644 index 0000000..3930dac Binary files /dev/null and b/train_temp/train_batch0.jpg differ diff --git a/train_temp/train_batch1.jpg b/train_temp/train_batch1.jpg new file mode 100644 index 0000000..4299887 Binary files /dev/null and b/train_temp/train_batch1.jpg differ diff --git a/train_temp/train_batch10.jpg b/train_temp/train_batch10.jpg new file mode 100644 index 0000000..81d393d Binary files /dev/null and b/train_temp/train_batch10.jpg differ diff --git a/train_temp/train_batch11.jpg b/train_temp/train_batch11.jpg new file mode 100644 index 0000000..0761130 Binary files /dev/null and b/train_temp/train_batch11.jpg differ diff --git a/train_temp/train_batch12.jpg b/train_temp/train_batch12.jpg new file mode 100644 index 0000000..01947fa Binary files /dev/null and b/train_temp/train_batch12.jpg differ diff --git a/train_temp/train_batch2.jpg b/train_temp/train_batch2.jpg new file mode 100644 index 0000000..a4a14d0 Binary files /dev/null and b/train_temp/train_batch2.jpg differ diff --git a/train_temp/val_batch0_labels.jpg b/train_temp/val_batch0_labels.jpg new file mode 100644 index 0000000..5c03436 Binary files /dev/null and b/train_temp/val_batch0_labels.jpg differ diff --git a/train_temp/val_batch0_pred.jpg b/train_temp/val_batch0_pred.jpg new file mode 100644 index 0000000..2cb3852 Binary files /dev/null and b/train_temp/val_batch0_pred.jpg differ diff --git a/train_temp/weights/best.pt b/train_temp/weights/best.pt new file mode 100644 index 0000000..402ccba Binary files /dev/null and b/train_temp/weights/best.pt differ diff --git a/train_temp/weights/last.pt b/train_temp/weights/last.pt new file mode 100644 index 0000000..bc0d75e Binary files /dev/null and b/train_temp/weights/last.pt differ