diff --git a/runs/detect/predict42/4.avi b/runs/detect/predict42/4.avi new file mode 100644 index 0000000..4cc1f34 Binary files /dev/null and b/runs/detect/predict42/4.avi differ diff --git a/runs/detect/predict43/3.avi b/runs/detect/predict43/3.avi new file mode 100644 index 0000000..598005e Binary files /dev/null and b/runs/detect/predict43/3.avi differ diff --git a/runs/detect/train30/results.csv b/runs/detect/train30/results.csv index 863d0e3..50f6f60 100644 --- a/runs/detect/train30/results.csv +++ b/runs/detect/train30/results.csv @@ -1,3 +1,33 @@ 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,11.1511,0.32252,0.30008,0.82477,0.70044,0.61111,0.70693,0.40802,1.4156,1.82764,1.29528,0.00012,0.00012,0.00012 2,22.5621,0.53184,0.36156,0.91688,0.72808,0.5958,0.70432,0.41349,1.33491,1.92438,1.2624,0.000259485,0.000259485,0.000259485 +3,37.4009,0.42627,0.30953,0.85093,0.82322,0.61111,0.66856,0.40661,1.30339,1.9308,1.246,0.000398416,0.000398416,0.000398416 +4,50.6053,0.3696,0.29421,0.72918,0.78449,0.607,0.63972,0.39453,1.32309,1.99121,1.23852,0.000536792,0.000536792,0.000536792 +5,61.9837,0.4193,0.31864,0.84078,0.79054,0.62988,0.69224,0.41965,1.29797,2.11731,1.23189,0.000674614,0.000674614,0.000674614 +6,72.3263,0.44745,0.33252,0.85214,0.77206,0.66667,0.69498,0.43195,1.30557,2.09769,1.25861,0.000811882,0.000811882,0.000811882 +7,82.5294,0.52376,0.36627,0.91823,1,0.61006,0.72617,0.43632,1.33534,2.20262,1.28968,0.000948595,0.000948595,0.000948595 +8,92.8043,0.45509,0.3499,0.87992,0.97725,0.5,0.62406,0.40349,1.51525,2.37502,1.42598,0.00108475,0.00108475,0.00108475 +9,103.013,0.48256,0.33764,0.84183,0.86737,0.44444,0.56804,0.37267,1.59608,2.44389,1.50879,0.00122036,0.00122036,0.00122036 +10,113.226,0.47736,0.35339,0.80598,0.76409,0.44444,0.50101,0.3135,1.79525,3.14093,1.60676,0.00135541,0.00135541,0.00135541 +11,123.477,0.50813,0.36954,0.90949,1,0.31529,0.41452,0.22007,1.94678,3.69518,1.63844,0.0014899,0.0014899,0.0014899 +12,133.811,0.5312,0.42649,0.88152,1,0.27254,0.42707,0.22986,1.68947,3.58509,1.4942,0.00162385,0.00162385,0.00162385 +13,144.038,0.48948,0.37137,0.90823,0.97182,0.27778,0.50051,0.27704,1.58752,3.28346,1.49017,0.00175723,0.00175723,0.00175723 +14,154.295,0.54969,0.39994,0.9252,0.97182,0.27778,0.50051,0.27704,1.58752,3.28346,1.49017,0.00189006,0.00189006,0.00189006 +15,164.571,0.55721,0.37372,0.90538,0.8482,0.27778,0.6294,0.31865,1.67169,3.17026,1.53199,0.00194456,0.00194456,0.00194456 +16,174.764,0.52692,0.42098,0.93484,1,0.3789,0.63598,0.32008,1.66792,3.03167,1.54914,0.0019406,0.0019406,0.0019406 +17,184.977,0.50085,0.39323,0.89325,0.87185,0.37908,0.60601,0.31689,1.6107,2.87598,1.49175,0.00193664,0.00193664,0.00193664 +18,195.244,0.5601,0.45519,0.88098,0.61825,0.5,0.51566,0.2881,1.72873,2.65876,1.59133,0.00193268,0.00193268,0.00193268 +19,205.494,0.52197,0.38114,0.88284,0.39224,0.5,0.2941,0.14758,1.82713,2.44741,1.58195,0.00192872,0.00192872,0.00192872 +20,215.77,0.51736,0.38366,0.89816,0.31353,0.55556,0.28387,0.12657,1.86683,2.18835,1.63153,0.00192476,0.00192476,0.00192476 +21,225.935,0.65567,0.43975,0.95891,0.63418,0.33333,0.35534,0.1414,2.12005,2.47744,1.72382,0.0019208,0.0019208,0.0019208 +22,236.063,0.47801,0.39977,0.89267,0.63418,0.33333,0.35534,0.1414,2.12005,2.47744,1.72382,0.00191684,0.00191684,0.00191684 +23,246.45,0.50685,0.41625,0.90029,0.57389,0.27778,0.31262,0.12226,2.18812,2.70416,1.73106,0.00191288,0.00191288,0.00191288 +24,256.707,0.58099,0.46448,0.89613,0.3001,0.26274,0.14864,0.06284,2.34146,3.34958,1.80192,0.00190892,0.00190892,0.00190892 +25,266.977,0.55089,0.41565,0.91733,0.16242,0.16667,0.0592,0.02662,2.40541,4.11065,1.78735,0.00190496,0.00190496,0.00190496 +26,277.276,0.65283,0.44754,0.92313,0.06544,0.05556,0.01788,0.0092,2.32548,4.59722,1.67037,0.001901,0.001901,0.001901 +27,287.498,0.45837,0.44487,0.7643,0.11447,0.16667,0.03717,0.01996,2.23564,4.46627,1.60797,0.00189704,0.00189704,0.00189704 +28,297.868,0.52012,0.4033,0.84711,0.6704,0.27778,0.28126,0.18306,1.91721,3.81855,1.49908,0.00189308,0.00189308,0.00189308 +29,308.225,0.56528,0.47869,0.88334,0.67392,0.44444,0.4585,0.25566,1.84635,3.22692,1.47325,0.00188912,0.00188912,0.00188912 +30,318.47,0.49045,0.50574,0.79166,0.67392,0.44444,0.4585,0.25566,1.84635,3.22692,1.47325,0.00188516,0.00188516,0.00188516 +31,328.679,0.53042,0.41121,0.90108,0.57392,0.38889,0.46928,0.28433,1.61208,3.19046,1.35925,0.0018812,0.0018812,0.0018812 +32,338.939,0.51546,0.46043,0.9401,0.53584,0.44444,0.47004,0.3152,1.52513,3.2269,1.2985,0.00187724,0.00187724,0.00187724 diff --git a/runs/detect/train30/weights/best.pt b/runs/detect/train30/weights/best.pt index 594cac0..7c33937 100644 Binary files a/runs/detect/train30/weights/best.pt and b/runs/detect/train30/weights/best.pt differ diff --git a/runs/detect/train30/weights/last.pt b/runs/detect/train30/weights/last.pt index 594cac0..414e091 100644 Binary files a/runs/detect/train30/weights/last.pt and b/runs/detect/train30/weights/last.pt differ diff --git a/runs/detect/train31/F1_curve.png b/runs/detect/train31/F1_curve.png new file mode 100644 index 0000000..026f0ea Binary files /dev/null and b/runs/detect/train31/F1_curve.png differ diff --git a/runs/detect/train31/PR_curve.png b/runs/detect/train31/PR_curve.png new file mode 100644 index 0000000..d3ed726 Binary files /dev/null and b/runs/detect/train31/PR_curve.png differ diff --git a/runs/detect/train31/P_curve.png b/runs/detect/train31/P_curve.png new file mode 100644 index 0000000..ffdab9f Binary files /dev/null and b/runs/detect/train31/P_curve.png differ diff --git a/runs/detect/train31/R_curve.png b/runs/detect/train31/R_curve.png new file mode 100644 index 0000000..08aa791 Binary files /dev/null and b/runs/detect/train31/R_curve.png differ diff --git a/runs/detect/train31/args.yaml b/runs/detect/train31/args.yaml new file mode 100644 index 0000000..799289c --- /dev/null +++ b/runs/detect/train31/args.yaml @@ -0,0 +1,105 @@ +task: detect +mode: train +model: C:\workspace\le-yolo\runs\detect\train30\weights\last.pt +data: data.yaml +epochs: 500 +time: null +patience: 100 +batch: 8 +imgsz: 640 +save: true +save_period: -1 +cache: false +device: cpu +workers: 8 +project: null +name: train31 +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 +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 +cutmix: 0.0 +copy_paste: 0.0 +copy_paste_mode: flip +auto_augment: randaugment +erasing: 0.4 +cfg: null +tracker: botsort.yaml +save_dir: C:\workspace\le-yolo\runs\detect\train31 diff --git a/runs/detect/train31/confusion_matrix.png b/runs/detect/train31/confusion_matrix.png new file mode 100644 index 0000000..076e106 Binary files /dev/null and b/runs/detect/train31/confusion_matrix.png differ diff --git a/runs/detect/train31/confusion_matrix_normalized.png b/runs/detect/train31/confusion_matrix_normalized.png new file mode 100644 index 0000000..6f15bcf Binary files /dev/null and b/runs/detect/train31/confusion_matrix_normalized.png differ diff --git a/runs/detect/train31/labels.jpg b/runs/detect/train31/labels.jpg new file mode 100644 index 0000000..cf7777b Binary files /dev/null and b/runs/detect/train31/labels.jpg differ diff --git a/runs/detect/train31/results.csv b/runs/detect/train31/results.csv new file mode 100644 index 0000000..58bafc2 --- /dev/null +++ b/runs/detect/train31/results.csv @@ -0,0 +1,104 @@ +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,11.7592,0.52287,0.39567,0.87938,0.69766,0.51353,0.61429,0.42203,1.50015,2.75363,1.27951,0.00012,0.00012,0.00012 +2,22.5396,0.6351,0.42801,0.93971,0.88976,0.55556,0.69116,0.46285,1.49408,2.2285,1.20662,0.000259485,0.000259485,0.000259485 +3,34.008,0.44252,0.35628,0.85151,0.7785,0.66667,0.74956,0.4684,1.53208,2.02281,1.19172,0.000398416,0.000398416,0.000398416 +4,54.198,0.38519,0.32572,0.71835,0.78249,0.66667,0.70686,0.37874,1.77729,2.30292,1.30834,0.000536792,0.000536792,0.000536792 +5,69.4036,0.47938,0.34475,0.84959,0.32609,0.83333,0.63968,0.33128,1.95667,3.08094,1.5152,0.000674614,0.000674614,0.000674614 +6,80.1716,0.51608,0.44141,0.87255,0.37396,0.38889,0.3627,0.19424,1.94175,3.66977,1.61073,0.000811882,0.000811882,0.000811882 +7,91.2062,0.47391,0.35984,0.90902,0.33473,0.22222,0.24196,0.14965,1.94508,3.89411,1.64544,0.000948595,0.000948595,0.000948595 +8,102.47,0.49139,0.37155,0.89161,0.09372,0.16667,0.09437,0.0752,2.05256,4.5331,1.78094,0.00108475,0.00108475,0.00108475 +9,113.521,0.44504,0.35109,0.83194,0.12894,0.23884,0.09614,0.07206,2.15068,4.64702,1.85204,0.00122036,0.00122036,0.00122036 +10,124.135,0.45647,0.34905,0.78584,0.25805,0.22222,0.15519,0.11716,2.07997,4.48882,1.7667,0.00135541,0.00135541,0.00135541 +11,134.784,0.50634,0.36281,0.88541,0.65761,0.21468,0.38209,0.23478,2.07326,4.45147,1.66727,0.0014899,0.0014899,0.0014899 +12,145.554,0.45616,0.3505,0.86203,0.28205,0.61111,0.41068,0.21365,2.17257,4.51248,1.7278,0.00162385,0.00162385,0.00162385 +13,157.032,0.45994,0.36961,0.89389,0.28205,0.61111,0.44699,0.20331,2.24113,4.5577,1.8735,0.00175723,0.00175723,0.00175723 +14,167.907,0.58812,0.38493,0.86773,0.28205,0.61111,0.44699,0.20331,2.24113,4.5577,1.8735,0.00189006,0.00189006,0.00189006 +15,179.574,0.49592,0.33203,0.82573,0.53918,0.38889,0.46054,0.18801,2.22688,4.07441,1.85708,0.00194456,0.00194456,0.00194456 +16,190.92,0.57167,0.38875,0.93089,1,0.27608,0.46465,0.22159,2.20176,3.40933,1.80533,0.0019406,0.0019406,0.0019406 +17,208.334,0.47897,0.39232,0.88036,0.58477,0.39138,0.47575,0.20757,2.20711,2.98279,1.72269,0.00193664,0.00193664,0.00193664 +18,227.885,0.62155,0.48452,0.87774,0.31516,0.33333,0.28736,0.1157,2.38244,2.93723,1.81084,0.00193268,0.00193268,0.00193268 +19,247.768,0.53804,0.39762,0.87266,0.194,0.27778,0.19975,0.07771,2.43879,3.06565,1.87056,0.00192872,0.00192872,0.00192872 +20,261.366,0.52722,0.37736,0.88552,0.18619,0.16667,0.15408,0.07231,2.59341,2.92617,2.03541,0.00192476,0.00192476,0.00192476 +21,272.341,0.62221,0.43519,0.93248,0.16133,0.32087,0.10856,0.04991,2.73894,3.02749,2.24268,0.0019208,0.0019208,0.0019208 +22,282.812,0.51288,0.38445,0.88448,0.16133,0.32087,0.10856,0.04991,2.73894,3.02749,2.24268,0.00191684,0.00191684,0.00191684 +23,293.699,0.52546,0.40857,0.8878,0.16948,0.16667,0.08555,0.05041,2.73324,3.03591,2.21266,0.00191288,0.00191288,0.00191288 +24,304.71,0.5468,0.4309,0.8818,0.24295,0.26786,0.15822,0.08532,2.65684,2.75586,2.05549,0.00190892,0.00190892,0.00190892 +25,315.626,0.52064,0.37387,0.90962,0.41756,0.33333,0.23666,0.14,2.45717,2.63435,1.83126,0.00190496,0.00190496,0.00190496 +26,326.208,0.64412,0.45156,0.92101,0.5338,0.38889,0.28401,0.14072,2.46925,2.7688,1.75905,0.001901,0.001901,0.001901 +27,336.86,0.43644,0.6765,0.75557,0.38046,0.38889,0.2583,0.14403,2.4169,3.3512,1.7964,0.00189704,0.00189704,0.00189704 +28,347.485,0.4973,0.38541,0.83598,0.48887,0.16667,0.18813,0.12854,2.35649,4.3374,2.07669,0.00189308,0.00189308,0.00189308 +29,358.682,0.55264,0.42661,0.89092,0.26264,0.16667,0.12392,0.09412,2.28123,5.38066,2.14752,0.00188912,0.00188912,0.00188912 +30,368.856,0.51334,0.38524,0.80745,0.26264,0.16667,0.12392,0.09412,2.28123,5.38066,2.14752,0.00188516,0.00188516,0.00188516 +31,379.047,0.52472,0.43753,0.90441,0.0914,0.05556,0.03315,0.02238,2.45677,5.76044,2.30373,0.0018812,0.0018812,0.0018812 +32,389.38,0.48927,0.38262,0.92498,0.00429,0.05556,0.0037,0.00259,2.43896,6.04092,2.37186,0.00187724,0.00187724,0.00187724 +33,399.896,0.50329,0.40521,0.90676,0.00373,0.11111,0.00268,0.00138,2.56368,5.99519,2.45869,0.00187328,0.00187328,0.00187328 +34,410.158,0.54834,0.42353,0.89736,0.0092,0.33333,0.00741,0.00382,2.46938,5.62497,2.38549,0.00186932,0.00186932,0.00186932 +35,420.421,0.52193,0.41558,0.87414,0.01653,0.05556,0.01507,0.0073,2.35841,4.84176,2.26095,0.00186536,0.00186536,0.00186536 +36,430.614,0.6594,0.50361,0.98472,0.04072,0.11111,0.01961,0.01035,2.3418,4.60929,2.25844,0.0018614,0.0018614,0.0018614 +37,440.831,0.54962,0.39433,0.91744,0.06406,0.11111,0.02837,0.01865,2.29659,4.49172,2.21217,0.00185744,0.00185744,0.00185744 +38,450.961,0.47958,0.37417,0.86916,0.06406,0.11111,0.02837,0.01865,2.29659,4.49172,2.21217,0.00185348,0.00185348,0.00185348 +39,461.115,0.47275,0.37939,0.85251,0.17498,0.11111,0.06364,0.04561,2.284,4.17177,2.02313,0.00184952,0.00184952,0.00184952 +40,471.613,0.51614,0.42277,0.89055,0.37855,0.22222,0.19808,0.15046,2.15678,3.7086,1.82675,0.00184556,0.00184556,0.00184556 +41,481.742,0.50416,0.39677,0.88655,0.54907,0.2713,0.23516,0.17307,2.18043,3.69748,1.89272,0.0018416,0.0018416,0.0018416 +42,491.857,0.45929,0.36052,0.88009,0.35212,0.33333,0.24908,0.17629,2.09017,3.56358,1.77953,0.00183764,0.00183764,0.00183764 +43,502.051,0.51473,0.36645,0.90245,0.45717,0.33333,0.24868,0.16953,2.12547,3.45162,1.75784,0.00183368,0.00183368,0.00183368 +44,512.17,0.54863,0.38077,0.9158,0.53812,0.33333,0.26222,0.17122,2.18998,3.3507,1.76629,0.00182972,0.00182972,0.00182972 +45,522.352,0.50288,0.39283,0.88992,0.5423,0.27778,0.21687,0.11739,2.4259,3.68169,1.95304,0.00182576,0.00182576,0.00182576 +46,532.479,0.77158,0.45683,1.17651,0.5423,0.27778,0.21687,0.11739,2.4259,3.68169,1.95304,0.0018218,0.0018218,0.0018218 +47,542.602,0.45378,0.35506,0.80192,0.50758,0.27778,0.23094,0.13026,2.28101,3.78436,1.87264,0.00181784,0.00181784,0.00181784 +48,552.761,0.58421,0.42721,0.93503,0.40299,0.27778,0.19086,0.10591,2.31434,3.6444,1.85348,0.00181388,0.00181388,0.00181388 +49,562.892,0.4468,0.34993,0.86569,0.31896,0.22222,0.13818,0.07723,2.45796,3.78633,2.01705,0.00180992,0.00180992,0.00180992 +50,573.001,0.70092,0.39964,1.16962,0.19186,0.22222,0.14671,0.0779,2.61296,3.83913,2.03691,0.00180596,0.00180596,0.00180596 +51,583.115,0.48673,0.37179,0.91225,0.19548,0.20317,0.12415,0.0594,2.48022,3.87955,1.90244,0.001802,0.001802,0.001802 +52,593.232,0.60108,0.44373,0.95932,0.16831,0.22222,0.1219,0.04677,2.41788,3.89216,1.88709,0.00179804,0.00179804,0.00179804 +53,603.388,0.47316,0.35967,0.87407,0.18237,0.34726,0.12883,0.0485,2.35549,3.72626,1.85734,0.00179408,0.00179408,0.00179408 +54,613.469,0.49431,0.34351,0.87688,0.18237,0.34726,0.12883,0.0485,2.35549,3.72626,1.85734,0.00179012,0.00179012,0.00179012 +55,623.633,0.62586,0.40334,1.02638,0.11947,0.22222,0.06932,0.02641,2.35661,4.25897,1.94806,0.00178616,0.00178616,0.00178616 +56,633.8,0.5306,0.37947,0.89266,0.15409,0.27778,0.11115,0.03594,2.26375,4.48295,1.91815,0.0017822,0.0017822,0.0017822 +57,643.985,0.53462,0.39051,0.90203,0.23597,0.16667,0.10659,0.04491,2.25071,4.4134,1.8892,0.00177824,0.00177824,0.00177824 +58,654.504,0.56209,0.38206,0.89792,0.20674,0.27778,0.12559,0.06854,2.27993,4.19742,1.92339,0.00177428,0.00177428,0.00177428 +59,664.64,0.54835,0.42767,0.93312,0.3039,0.33333,0.1701,0.08765,2.23335,4.12682,1.87014,0.00177032,0.00177032,0.00177032 +60,674.78,0.43443,0.41911,0.758,0.25381,0.2837,0.13655,0.06516,2.1684,4.35749,1.80319,0.00176636,0.00176636,0.00176636 +61,685.01,0.51045,0.36135,0.86086,0.28432,0.26544,0.15058,0.06768,2.01662,4.5956,1.72649,0.0017624,0.0017624,0.0017624 +62,695.109,0.54765,0.41848,0.93568,0.28432,0.26544,0.15058,0.06768,2.01662,4.5956,1.72649,0.00175844,0.00175844,0.00175844 +63,705.248,0.44609,0.36378,0.78576,0.25566,0.28659,0.19,0.07949,2.00559,4.46423,1.67202,0.00175448,0.00175448,0.00175448 +64,715.381,0.52945,0.40027,0.93619,0.2443,0.11111,0.14122,0.06915,2.02919,4.6024,1.72874,0.00175052,0.00175052,0.00175052 +65,725.466,0.46488,0.34765,0.84152,0.18917,0.11111,0.11267,0.05522,2.33121,4.74077,1.86056,0.00174656,0.00174656,0.00174656 +66,735.571,0.52296,0.38556,0.8607,0.31611,0.15518,0.16826,0.06057,2.42948,4.78893,1.91481,0.0017426,0.0017426,0.0017426 +67,745.692,0.53151,0.38152,0.87026,0.27859,0.38889,0.21773,0.07661,2.32452,4.55612,1.83225,0.00173864,0.00173864,0.00173864 +68,755.907,0.51558,0.36575,0.89566,0.52398,0.16667,0.29648,0.09998,2.3045,4.32543,1.82168,0.00173468,0.00173468,0.00173468 +69,766.004,0.52092,0.40619,0.88941,0.46853,0.16667,0.3116,0.107,2.25486,4.23041,1.76153,0.00173072,0.00173072,0.00173072 +70,776.066,0.47036,0.39209,0.79645,0.46853,0.16667,0.3116,0.107,2.25486,4.23041,1.76153,0.00172676,0.00172676,0.00172676 +71,786.183,0.57067,0.41829,0.90494,0.52686,0.27778,0.38981,0.14691,2.11311,3.83055,1.68418,0.0017228,0.0017228,0.0017228 +72,796.302,0.44272,0.36177,0.89767,0.57149,0.22222,0.33204,0.14387,2.08719,3.74349,1.70011,0.00171884,0.00171884,0.00171884 +73,806.432,0.51153,0.35831,0.88658,0.78299,0.27778,0.40739,0.13615,2.11226,3.77238,1.72693,0.00171488,0.00171488,0.00171488 +74,816.536,0.49449,0.36477,0.92266,0.73794,0.31457,0.3834,0.10436,2.31183,3.71107,1.81314,0.00171092,0.00171092,0.00171092 +75,826.734,0.44293,0.33484,0.84142,0.59642,0.33333,0.42324,0.0996,2.24754,3.30195,1.81964,0.00170696,0.00170696,0.00170696 +76,836.808,0.53011,0.37554,0.85507,0.76739,0.33333,0.39164,0.10129,2.08938,3.00342,1.7105,0.001703,0.001703,0.001703 +77,846.917,0.49359,0.35962,0.85989,0.33742,0.16667,0.1104,0.05373,2.34424,3.22778,1.90942,0.00169904,0.00169904,0.00169904 +78,856.988,0.51938,0.37536,0.89305,0.33742,0.16667,0.1104,0.05373,2.34424,3.22778,1.90942,0.00169508,0.00169508,0.00169508 +79,867.089,0.56197,0.35926,0.88821,0.16177,0.11111,0.06311,0.02728,2.70732,3.8661,2.16785,0.00169112,0.00169112,0.00169112 +80,877.485,0.50077,0.32886,0.83741,0.28438,0.11111,0.07544,0.03654,2.80138,3.56308,2.22223,0.00168716,0.00168716,0.00168716 +81,888.012,0.49802,0.37266,0.91998,0.25729,0.11111,0.08745,0.04427,2.74391,3.50635,2.15866,0.0016832,0.0016832,0.0016832 +82,898.523,0.4531,0.34236,0.89177,0.16129,0.22222,0.10805,0.04468,2.63815,3.39929,2.075,0.00167924,0.00167924,0.00167924 +83,908.867,0.51251,0.39236,0.89988,0.21907,0.33333,0.17122,0.05178,2.5091,3.23449,1.9564,0.00167528,0.00167528,0.00167528 +84,919.261,0.47108,0.39815,0.86693,0.22961,0.33333,0.1801,0.05071,2.50719,3.53065,1.94488,0.00167132,0.00167132,0.00167132 +85,929.834,0.54385,0.40752,0.92181,0.4805,0.16667,0.18001,0.04918,2.59303,3.81146,1.9831,0.00166736,0.00166736,0.00166736 +86,941.126,0.46109,0.35186,0.87754,0.4805,0.16667,0.18001,0.04918,2.59303,3.81146,1.9831,0.0016634,0.0016634,0.0016634 +87,951.972,0.50057,0.35057,0.91015,0.61599,0.11111,0.17343,0.03868,2.77947,4.02777,2.08747,0.00165944,0.00165944,0.00165944 +88,962.692,0.45603,0.33199,0.86677,0.8487,0.11111,0.15665,0.02795,2.97186,3.79906,2.23161,0.00165548,0.00165548,0.00165548 +89,973.087,0.43307,0.32023,0.86535,0.36287,0.16667,0.18773,0.0253,3.04482,3.90671,2.43114,0.00165152,0.00165152,0.00165152 +90,983.813,0.5287,0.35963,0.87884,0.55284,0.16667,0.19318,0.02424,2.98667,4.20034,2.44746,0.00164756,0.00164756,0.00164756 +91,994.578,0.56352,0.39514,0.86273,0.75184,0.11111,0.12601,0.01594,2.89386,4.78159,2.36998,0.0016436,0.0016436,0.0016436 +92,1004.97,0.46157,0.33644,0.90895,0.4589,0.16667,0.14082,0.02461,2.81949,4.94958,2.28738,0.00163964,0.00163964,0.00163964 +93,1015.28,0.50549,0.36989,0.92968,0.69408,0.16667,0.17861,0.03799,2.86283,4.47323,2.16909,0.00163568,0.00163568,0.00163568 +94,1026.64,0.54453,0.36556,0.91366,0.69408,0.16667,0.17861,0.03799,2.86283,4.47323,2.16909,0.00163172,0.00163172,0.00163172 +95,1037.42,0.45418,0.34127,0.8731,0.4394,0.16667,0.16343,0.03677,2.84144,5.09985,2.25183,0.00162776,0.00162776,0.00162776 +96,1047.91,0.48752,0.37778,0.92677,0.22008,0.16667,0.08745,0.01539,2.74118,6.24654,2.29197,0.0016238,0.0016238,0.0016238 +97,1058.91,0.4287,0.34157,0.85858,0.09371,0.16667,0.02972,0.00355,2.74577,7.17663,2.32418,0.00161984,0.00161984,0.00161984 +98,1070.12,0.45141,0.39729,0.87353,0.02246,0.05556,0.00621,0.00096,2.8139,7.2233,2.43831,0.00161588,0.00161588,0.00161588 +99,1086.65,0.50609,0.38367,0.92804,0.00405,0.05556,0.00219,0.00132,2.86479,6.70936,2.46255,0.00161192,0.00161192,0.00161192 +100,1106.09,0.49679,0.40437,0.93468,0.26282,0.16667,0.09513,0.01479,2.91953,4.73198,2.40587,0.00160796,0.00160796,0.00160796 +101,1117.63,0.43488,0.33616,0.79537,0.23809,0.33333,0.18919,0.03903,2.81967,4.29117,2.38804,0.001604,0.001604,0.001604 +102,1127.87,0.49432,0.38127,0.88334,0.23809,0.33333,0.18919,0.03903,2.81967,4.29117,2.38804,0.00160004,0.00160004,0.00160004 +103,1139.05,0.53965,0.38016,0.84544,0.41177,0.22222,0.20703,0.03581,2.82203,4.28733,2.38276,0.00159608,0.00159608,0.00159608 diff --git a/runs/detect/train31/results.png b/runs/detect/train31/results.png new file mode 100644 index 0000000..fd1c38b Binary files /dev/null and b/runs/detect/train31/results.png differ diff --git a/runs/detect/train31/train_batch0.jpg b/runs/detect/train31/train_batch0.jpg new file mode 100644 index 0000000..ba7ea35 Binary files /dev/null and b/runs/detect/train31/train_batch0.jpg differ diff --git a/runs/detect/train31/train_batch1.jpg b/runs/detect/train31/train_batch1.jpg new file mode 100644 index 0000000..ebc5faf Binary files /dev/null and b/runs/detect/train31/train_batch1.jpg differ diff --git a/runs/detect/train31/train_batch2.jpg b/runs/detect/train31/train_batch2.jpg new file mode 100644 index 0000000..c2a5bd8 Binary files /dev/null and b/runs/detect/train31/train_batch2.jpg differ diff --git a/runs/detect/train31/val_batch0_labels.jpg b/runs/detect/train31/val_batch0_labels.jpg new file mode 100644 index 0000000..9262645 Binary files /dev/null and b/runs/detect/train31/val_batch0_labels.jpg differ diff --git a/runs/detect/train31/val_batch0_pred.jpg b/runs/detect/train31/val_batch0_pred.jpg new file mode 100644 index 0000000..d2fa60c Binary files /dev/null and b/runs/detect/train31/val_batch0_pred.jpg differ diff --git a/runs/detect/train31/val_batch1_labels.jpg b/runs/detect/train31/val_batch1_labels.jpg new file mode 100644 index 0000000..12d1041 Binary files /dev/null and b/runs/detect/train31/val_batch1_labels.jpg differ diff --git a/runs/detect/train31/val_batch1_pred.jpg b/runs/detect/train31/val_batch1_pred.jpg new file mode 100644 index 0000000..63dda61 Binary files /dev/null and b/runs/detect/train31/val_batch1_pred.jpg differ diff --git a/runs/detect/train31/weights/best.pt b/runs/detect/train31/weights/best.pt new file mode 100644 index 0000000..eb17fa4 Binary files /dev/null and b/runs/detect/train31/weights/best.pt differ diff --git a/runs/detect/train31/weights/last.pt b/runs/detect/train31/weights/last.pt new file mode 100644 index 0000000..ffb13a2 Binary files /dev/null and b/runs/detect/train31/weights/last.pt differ diff --git a/runs/detect/train312/F1_curve.png b/runs/detect/train312/F1_curve.png new file mode 100644 index 0000000..026f0ea Binary files /dev/null and b/runs/detect/train312/F1_curve.png differ diff --git a/runs/detect/train312/PR_curve.png b/runs/detect/train312/PR_curve.png new file mode 100644 index 0000000..d3ed726 Binary files /dev/null and b/runs/detect/train312/PR_curve.png differ diff --git a/runs/detect/train312/P_curve.png b/runs/detect/train312/P_curve.png new file mode 100644 index 0000000..ffdab9f Binary files /dev/null and b/runs/detect/train312/P_curve.png differ diff --git a/runs/detect/train312/R_curve.png b/runs/detect/train312/R_curve.png new file mode 100644 index 0000000..08aa791 Binary files /dev/null and b/runs/detect/train312/R_curve.png differ diff --git a/runs/detect/train312/confusion_matrix.png b/runs/detect/train312/confusion_matrix.png new file mode 100644 index 0000000..076e106 Binary files /dev/null and b/runs/detect/train312/confusion_matrix.png differ diff --git a/runs/detect/train312/confusion_matrix_normalized.png b/runs/detect/train312/confusion_matrix_normalized.png new file mode 100644 index 0000000..6f15bcf Binary files /dev/null and b/runs/detect/train312/confusion_matrix_normalized.png differ diff --git a/runs/detect/train312/val_batch0_labels.jpg b/runs/detect/train312/val_batch0_labels.jpg new file mode 100644 index 0000000..2d9a335 Binary files /dev/null and b/runs/detect/train312/val_batch0_labels.jpg differ diff --git a/runs/detect/train312/val_batch0_pred.jpg b/runs/detect/train312/val_batch0_pred.jpg new file mode 100644 index 0000000..c14278e Binary files /dev/null and b/runs/detect/train312/val_batch0_pred.jpg differ diff --git a/runs/detect/train312/val_batch1_labels.jpg b/runs/detect/train312/val_batch1_labels.jpg new file mode 100644 index 0000000..085a63e Binary files /dev/null and b/runs/detect/train312/val_batch1_labels.jpg differ diff --git a/runs/detect/train312/val_batch1_pred.jpg b/runs/detect/train312/val_batch1_pred.jpg new file mode 100644 index 0000000..6d1158b Binary files /dev/null and b/runs/detect/train312/val_batch1_pred.jpg differ diff --git a/runs/detect/train312/val_batch2_labels.jpg b/runs/detect/train312/val_batch2_labels.jpg new file mode 100644 index 0000000..12d1041 Binary files /dev/null and b/runs/detect/train312/val_batch2_labels.jpg differ diff --git a/runs/detect/train312/val_batch2_pred.jpg b/runs/detect/train312/val_batch2_pred.jpg new file mode 100644 index 0000000..63dda61 Binary files /dev/null and b/runs/detect/train312/val_batch2_pred.jpg differ diff --git a/runs/detect/train32/F1_curve.png b/runs/detect/train32/F1_curve.png new file mode 100644 index 0000000..4e77acb Binary files /dev/null and b/runs/detect/train32/F1_curve.png differ diff --git a/runs/detect/train32/PR_curve.png b/runs/detect/train32/PR_curve.png new file mode 100644 index 0000000..3a63de7 Binary files /dev/null and b/runs/detect/train32/PR_curve.png differ diff --git a/runs/detect/train32/P_curve.png b/runs/detect/train32/P_curve.png new file mode 100644 index 0000000..c504a0c Binary files /dev/null and b/runs/detect/train32/P_curve.png differ diff --git a/runs/detect/train32/R_curve.png b/runs/detect/train32/R_curve.png new file mode 100644 index 0000000..928410f Binary files /dev/null and b/runs/detect/train32/R_curve.png differ diff --git a/runs/detect/train32/args.yaml b/runs/detect/train32/args.yaml new file mode 100644 index 0000000..7e85762 --- /dev/null +++ b/runs/detect/train32/args.yaml @@ -0,0 +1,105 @@ +task: detect +mode: train +model: C:\workspace\le-yolo\runs\detect\train30\weights\last.pt +data: data.yaml +epochs: 500 +time: null +patience: 100 +batch: 8 +imgsz: 640 +save: true +save_period: -1 +cache: false +device: cpu +workers: 8 +project: null +name: train32 +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 +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 +cutmix: 0.0 +copy_paste: 0.0 +copy_paste_mode: flip +auto_augment: randaugment +erasing: 0.4 +cfg: null +tracker: botsort.yaml +save_dir: C:\workspace\le-yolo\runs\detect\train32 diff --git a/runs/detect/train32/confusion_matrix.png b/runs/detect/train32/confusion_matrix.png new file mode 100644 index 0000000..3d8ba2b Binary files /dev/null and b/runs/detect/train32/confusion_matrix.png differ diff --git a/runs/detect/train32/confusion_matrix_normalized.png b/runs/detect/train32/confusion_matrix_normalized.png new file mode 100644 index 0000000..ab149b8 Binary files /dev/null and b/runs/detect/train32/confusion_matrix_normalized.png differ diff --git a/runs/detect/train32/labels.jpg b/runs/detect/train32/labels.jpg new file mode 100644 index 0000000..6cc3464 Binary files /dev/null and b/runs/detect/train32/labels.jpg differ diff --git a/runs/detect/train32/results.csv b/runs/detect/train32/results.csv new file mode 100644 index 0000000..dac8c8c --- /dev/null +++ b/runs/detect/train32/results.csv @@ -0,0 +1,102 @@ +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,7.63907,2.70917,4.49407,2.02465,0.99912,0.97872,0.97747,0.91819,0.41838,0.36787,0.86558,4e-05,4e-05,4e-05 +2,14.7997,2.22765,4.25443,1.75345,0.99881,0.97872,0.97724,0.90882,0.40916,0.35668,0.85675,9.9802e-05,9.9802e-05,9.9802e-05 +3,21.8956,1.79894,2.2375,1.3571,0.99858,0.97872,0.97686,0.90678,0.41569,0.34502,0.85603,0.000159366,0.000159366,0.000159366 +4,28.9274,1.732,1.71605,1.32749,0.99702,0.97872,0.97625,0.9021,0.41691,0.3773,0.85139,0.000218693,0.000218693,0.000218693 +5,36.0726,1.51583,1.2176,1.22265,0.99167,0.97872,0.97602,0.8814,0.4486,0.42174,0.86108,0.000277782,0.000277782,0.000277782 +6,42.9992,1.50008,1.20603,1.20527,0.9715,0.97872,0.97525,0.84578,0.57006,0.55669,0.91396,0.000336634,0.000336634,0.000336634 +7,50.1607,1.17856,1.06457,1.0255,0.94974,0.97872,0.97471,0.83043,0.66164,0.59542,0.978,0.000395248,0.000395248,0.000395248 +8,57.3054,1.04042,0.85323,0.99251,0.94447,0.97872,0.97505,0.83987,0.72603,0.59996,1.03553,0.000453624,0.000453624,0.000453624 +9,64.2003,1.164,0.94852,1.02117,0.92959,0.97872,0.9733,0.84751,0.76872,0.58319,1.07821,0.000511763,0.000511763,0.000511763 +10,71.247,1.26369,0.86267,1.24208,1,0.97339,0.97545,0.83919,0.77765,0.59624,1.10428,0.000569664,0.000569664,0.000569664 +11,78.0823,1.03984,0.73627,0.98369,1,0.95536,0.97506,0.84243,0.75775,0.57626,1.10348,0.000627328,0.000627328,0.000627328 +12,85.2249,1.25405,0.85407,1.0554,0.99278,0.95745,0.97516,0.84042,0.75268,0.55745,1.11194,0.000684754,0.000684754,0.000684754 +13,92.1208,0.97203,0.72676,1.05147,0.99723,0.97872,0.97575,0.84047,0.7736,0.53483,1.11246,0.000741942,0.000741942,0.000741942 +14,99.0156,1.01532,0.75534,1.00374,0.99693,0.97872,0.976,0.83682,0.82381,0.55106,1.14222,0.000798893,0.000798893,0.000798893 +15,105.913,0.94213,0.67325,0.97016,0.99693,0.97872,0.976,0.83682,0.82381,0.55106,1.14222,0.000855606,0.000855606,0.000855606 +16,113.322,0.79714,0.51982,0.89325,0.99667,0.97872,0.97631,0.79911,0.89067,0.60695,1.21469,0.000912082,0.000912082,0.000912082 +17,120.293,1.02492,0.61514,0.97705,1,0.97682,0.97648,0.80457,0.98069,0.65745,1.29521,0.00096832,0.00096832,0.00096832 +18,127.201,0.92379,0.67928,0.93293,1,0.97682,0.97648,0.80457,0.98069,0.65745,1.29521,0.00102432,0.00102432,0.00102432 +19,134.089,0.89685,0.61271,0.90453,0.99362,0.97872,0.97673,0.79301,1.09955,0.77412,1.40548,0.00108008,0.00108008,0.00108008 +20,140.992,0.98846,0.64352,0.98551,0.99277,0.97872,0.97707,0.79259,1.16372,0.76491,1.45391,0.00113561,0.00113561,0.00113561 +21,147.822,0.97386,0.74687,0.97902,0.99277,0.97872,0.97707,0.79259,1.16372,0.76491,1.45391,0.0011909,0.0011909,0.0011909 +22,154.669,1.00683,0.60883,0.9834,0.99692,0.95745,0.97743,0.7862,1.23137,0.68919,1.49529,0.00124595,0.00124595,0.00124595 +23,161.483,0.91046,0.64467,0.99744,0.99692,0.95745,0.97743,0.7862,1.23137,0.68919,1.49529,0.00130076,0.00130076,0.00130076 +24,168.57,0.84125,0.59386,0.89692,0.93849,0.97396,0.97619,0.76515,1.3093,0.64991,1.50103,0.00135533,0.00135533,0.00135533 +25,175.564,0.8557,0.59673,0.8886,0.93849,0.97396,0.97619,0.76515,1.3093,0.64991,1.50103,0.00140967,0.00140967,0.00140967 +26,182.415,1.05203,0.66896,1.03143,0.97302,0.91489,0.97423,0.7418,1.38855,0.63057,1.54444,0.00146377,0.00146377,0.00146377 +27,189.245,0.8033,0.62459,0.91053,0.97302,0.91489,0.97423,0.7418,1.38855,0.63057,1.54444,0.00151763,0.00151763,0.00151763 +28,196.121,0.89312,0.59819,0.88909,0.95354,0.91489,0.97145,0.7324,1.46177,0.64943,1.60919,0.00157126,0.00157126,0.00157126 +29,203.594,0.85721,0.56653,0.93244,0.95354,0.91489,0.97145,0.7324,1.46177,0.64943,1.60919,0.00162464,0.00162464,0.00162464 +30,210.617,0.9005,0.62716,0.94168,0.95354,0.91489,0.97145,0.7324,1.46177,0.64943,1.60919,0.00167779,0.00167779,0.00167779 +31,217.476,0.79788,0.62076,0.93894,0.96733,0.85106,0.9625,0.72508,1.51965,0.73935,1.72766,0.0017307,0.0017307,0.0017307 +32,224.986,0.77564,0.57184,0.92898,0.96733,0.85106,0.9625,0.72508,1.51965,0.73935,1.72766,0.00178338,0.00178338,0.00178338 +33,233.051,0.80249,0.51501,0.91057,0.97545,0.84563,0.94296,0.6827,1.55516,0.90677,1.89134,0.00183581,0.00183581,0.00183581 +34,241.105,0.83376,0.55029,0.89974,0.97545,0.84563,0.94296,0.6827,1.55516,0.90677,1.89134,0.00186932,0.00186932,0.00186932 +35,249.696,0.72198,0.51049,0.86315,0.97545,0.84563,0.94296,0.6827,1.55516,0.90677,1.89134,0.00186536,0.00186536,0.00186536 +36,258.063,0.81764,0.54125,0.91989,0.97378,0.82979,0.93639,0.6954,1.60172,0.92111,1.99599,0.0018614,0.0018614,0.0018614 +37,266.67,0.76624,0.5811,0.87194,0.97378,0.82979,0.93639,0.6954,1.60172,0.92111,1.99599,0.00185744,0.00185744,0.00185744 +38,275.247,0.70616,0.52048,0.83692,0.97378,0.82979,0.93639,0.6954,1.60172,0.92111,1.99599,0.00185348,0.00185348,0.00185348 +39,283.268,0.80025,0.57801,0.87308,0.97433,0.8077,0.9029,0.65711,1.63515,0.81447,1.95222,0.00184952,0.00184952,0.00184952 +40,290.959,0.84311,0.61695,0.99219,0.97433,0.8077,0.9029,0.65711,1.63515,0.81447,1.95222,0.00184556,0.00184556,0.00184556 +41,298.794,0.7594,0.52677,0.89593,0.9522,0.84767,0.91945,0.67852,1.60546,0.7564,1.93401,0.0018416,0.0018416,0.0018416 +42,306.783,0.82036,0.59852,0.97176,0.9522,0.84767,0.91945,0.67852,1.60546,0.7564,1.93401,0.00183764,0.00183764,0.00183764 +43,314.609,0.8502,0.61643,0.93547,0.9522,0.84767,0.91945,0.67852,1.60546,0.7564,1.93401,0.00183368,0.00183368,0.00183368 +44,322.343,0.74427,0.56837,0.89203,0.97453,0.81402,0.8989,0.67481,1.55939,0.76543,1.96081,0.00182972,0.00182972,0.00182972 +45,330.17,0.6368,0.47688,0.85325,0.97453,0.81402,0.8989,0.67481,1.55939,0.76543,1.96081,0.00182576,0.00182576,0.00182576 +46,337.966,0.69582,0.52199,0.87481,0.97453,0.81402,0.8989,0.67481,1.55939,0.76543,1.96081,0.0018218,0.0018218,0.0018218 +47,346.133,0.73562,0.54278,0.87558,0.94264,0.80851,0.88646,0.66755,1.53789,0.74703,1.97612,0.00181784,0.00181784,0.00181784 +48,352.997,0.77988,0.55341,0.90458,0.94264,0.80851,0.88646,0.66755,1.53789,0.74703,1.97612,0.00181388,0.00181388,0.00181388 +49,359.854,0.64288,0.45365,0.85448,0.94467,0.80851,0.8811,0.66314,1.56657,0.77116,1.95987,0.00180992,0.00180992,0.00180992 +50,366.72,0.6835,0.49352,0.85955,0.94467,0.80851,0.8811,0.66314,1.56657,0.77116,1.95987,0.00180596,0.00180596,0.00180596 +51,373.578,0.70843,0.49494,0.85846,0.94467,0.80851,0.8811,0.66314,1.56657,0.77116,1.95987,0.001802,0.001802,0.001802 +52,380.429,0.8536,0.59141,0.89611,0.92626,0.80194,0.87514,0.65763,1.60355,0.78128,1.9507,0.00179804,0.00179804,0.00179804 +53,387.374,0.81654,0.5779,0.93825,0.92626,0.80194,0.87514,0.65763,1.60355,0.78128,1.9507,0.00179408,0.00179408,0.00179408 +54,394.604,0.74893,0.52772,0.86002,0.92626,0.80194,0.87514,0.65763,1.60355,0.78128,1.9507,0.00179012,0.00179012,0.00179012 +55,401.609,0.88735,0.66636,0.96151,0.9681,0.78723,0.88236,0.64213,1.63055,0.80361,1.93165,0.00178616,0.00178616,0.00178616 +56,408.55,0.75877,0.52875,0.87482,0.9681,0.78723,0.88236,0.64213,1.63055,0.80361,1.93165,0.0017822,0.0017822,0.0017822 +57,415.448,0.7728,0.55349,0.88697,0.96913,0.76596,0.87738,0.64668,1.53719,0.8137,1.79539,0.00177824,0.00177824,0.00177824 +58,422.685,0.7351,0.53766,0.87352,0.96913,0.76596,0.87738,0.64668,1.53719,0.8137,1.79539,0.00177428,0.00177428,0.00177428 +59,430.553,0.76144,0.59656,0.85218,0.96913,0.76596,0.87738,0.64668,1.53719,0.8137,1.79539,0.00177032,0.00177032,0.00177032 +60,438.936,0.76499,0.57114,0.86835,0.92767,0.76596,0.88565,0.64631,1.5512,0.83701,1.76876,0.00176636,0.00176636,0.00176636 +61,446.435,0.66382,0.56661,0.82241,0.92767,0.76596,0.88565,0.64631,1.5512,0.83701,1.76876,0.0017624,0.0017624,0.0017624 +62,453.495,0.68367,0.47688,0.89085,0.92767,0.76596,0.88565,0.64631,1.5512,0.83701,1.76876,0.00175844,0.00175844,0.00175844 +63,460.658,0.68854,0.49969,0.9078,0.8042,0.89362,0.90818,0.60688,1.67533,0.86673,1.85942,0.00175448,0.00175448,0.00175448 +64,467.71,0.78518,0.59558,0.95986,0.8042,0.89362,0.90818,0.60688,1.67533,0.86673,1.85942,0.00175052,0.00175052,0.00175052 +65,474.809,0.67974,0.47885,0.86958,0.79064,0.89362,0.90476,0.59039,1.70069,0.90242,1.91292,0.00174656,0.00174656,0.00174656 +66,481.722,0.71374,0.51359,0.85046,0.79064,0.89362,0.90476,0.59039,1.70069,0.90242,1.91292,0.0017426,0.0017426,0.0017426 +67,489.399,0.79765,0.53107,0.89975,0.79064,0.89362,0.90476,0.59039,1.70069,0.90242,1.91292,0.00173864,0.00173864,0.00173864 +68,497.148,0.70784,0.49291,0.89467,0.82587,0.80851,0.88682,0.60732,1.64151,0.97339,1.95122,0.00173468,0.00173468,0.00173468 +69,505.093,0.83998,0.55884,0.89796,0.82587,0.80851,0.88682,0.60732,1.64151,0.97339,1.95122,0.00173072,0.00173072,0.00173072 +70,512.854,0.63445,0.42248,0.83863,0.82587,0.80851,0.88682,0.60732,1.64151,0.97339,1.95122,0.00172676,0.00172676,0.00172676 +71,520.763,0.70379,0.49597,0.86819,0.84016,0.80851,0.86679,0.59617,1.5825,1.0674,1.95819,0.0017228,0.0017228,0.0017228 +72,528.402,0.78343,0.52781,0.86032,0.84016,0.80851,0.86679,0.59617,1.5825,1.0674,1.95819,0.00171884,0.00171884,0.00171884 +73,536.215,0.65805,0.49055,0.87537,0.7579,0.79937,0.83955,0.58596,1.60346,1.14645,1.99698,0.00171488,0.00171488,0.00171488 +74,544.016,0.7471,0.54277,0.8316,0.7579,0.79937,0.83955,0.58596,1.60346,1.14645,1.99698,0.00171092,0.00171092,0.00171092 +75,551.8,0.74852,0.53142,0.88043,0.7579,0.79937,0.83955,0.58596,1.60346,1.14645,1.99698,0.00170696,0.00170696,0.00170696 +76,559.707,0.69011,0.46713,0.85847,0.74698,0.76596,0.83512,0.5692,1.57461,1.17368,1.99194,0.001703,0.001703,0.001703 +77,567.299,0.72188,0.51026,0.90831,0.74698,0.76596,0.83512,0.5692,1.57461,1.17368,1.99194,0.00169904,0.00169904,0.00169904 +78,575.002,0.75518,0.50844,0.9042,0.74698,0.76596,0.83512,0.5692,1.57461,1.17368,1.99194,0.00169508,0.00169508,0.00169508 +79,582.647,0.73466,0.50325,0.90463,0.76239,0.75104,0.84273,0.56276,1.59924,1.16856,1.96152,0.00169112,0.00169112,0.00169112 +80,590.361,0.64651,0.47375,0.84126,0.76239,0.75104,0.84273,0.56276,1.59924,1.16856,1.96152,0.00168716,0.00168716,0.00168716 +81,598.147,0.73129,0.49409,0.92695,0.76884,0.74468,0.81936,0.53968,1.63233,1.20549,1.95335,0.0016832,0.0016832,0.0016832 +82,606.268,0.65436,0.47217,0.86174,0.76884,0.74468,0.81936,0.53968,1.63233,1.20549,1.95335,0.00167924,0.00167924,0.00167924 +83,613.378,0.68755,0.46353,0.80133,0.76884,0.74468,0.81936,0.53968,1.63233,1.20549,1.95335,0.00167528,0.00167528,0.00167528 +84,620.24,0.75798,0.5305,0.94699,0.72636,0.78723,0.80995,0.51352,1.63355,1.23102,1.98695,0.00167132,0.00167132,0.00167132 +85,627.073,0.58781,0.42422,0.8475,0.72636,0.78723,0.80995,0.51352,1.63355,1.23102,1.98695,0.00166736,0.00166736,0.00166736 +86,633.956,0.6749,0.49486,0.83143,0.72636,0.78723,0.80995,0.51352,1.63355,1.23102,1.98695,0.0016634,0.0016634,0.0016634 +87,640.967,0.76251,0.50619,0.86481,0.69328,0.76953,0.77858,0.50373,1.62916,1.30002,1.99966,0.00165944,0.00165944,0.00165944 +88,647.776,0.61438,0.4451,0.87131,0.69328,0.76953,0.77858,0.50373,1.62916,1.30002,1.99966,0.00165548,0.00165548,0.00165548 +89,654.77,0.56289,0.4546,0.82389,0.7315,0.74468,0.77885,0.46848,1.67011,1.36403,2.04133,0.00165152,0.00165152,0.00165152 +90,661.973,0.57873,0.40353,0.85035,0.7315,0.74468,0.77885,0.46848,1.67011,1.36403,2.04133,0.00164756,0.00164756,0.00164756 +91,669.077,0.66058,0.51081,0.84518,0.7315,0.74468,0.77885,0.46848,1.67011,1.36403,2.04133,0.0016436,0.0016436,0.0016436 +92,676.094,0.67254,0.51756,0.86201,0.78409,0.74468,0.78603,0.4469,1.72356,1.38979,2.08012,0.00163964,0.00163964,0.00163964 +93,682.941,0.67494,0.49047,0.91852,0.78409,0.74468,0.78603,0.4469,1.72356,1.38979,2.08012,0.00163568,0.00163568,0.00163568 +94,689.953,0.67352,0.4834,0.85694,0.78409,0.74468,0.78603,0.4469,1.72356,1.38979,2.08012,0.00163172,0.00163172,0.00163172 +95,696.912,0.67889,0.46032,0.85274,0.8203,0.74468,0.812,0.45954,1.77089,1.40879,2.16685,0.00162776,0.00162776,0.00162776 +96,703.93,0.6799,0.47526,0.8715,0.8203,0.74468,0.812,0.45954,1.77089,1.40879,2.16685,0.0016238,0.0016238,0.0016238 +97,711.212,0.55019,0.43748,0.8411,0.82045,0.78723,0.82938,0.48868,1.80145,1.5016,2.26065,0.00161984,0.00161984,0.00161984 +98,718.621,0.6155,0.51893,0.86594,0.82045,0.78723,0.82938,0.48868,1.80145,1.5016,2.26065,0.00161588,0.00161588,0.00161588 +99,725.83,0.57495,0.45232,0.86671,0.82045,0.78723,0.82938,0.48868,1.80145,1.5016,2.26065,0.00161192,0.00161192,0.00161192 +100,733.002,0.58836,0.42686,0.84293,0.85389,0.78723,0.82661,0.49916,1.82295,1.49637,2.35177,0.00160796,0.00160796,0.00160796 +101,739.919,0.76866,0.49956,0.93021,0.85389,0.78723,0.82661,0.49916,1.82295,1.49637,2.35177,0.001604,0.001604,0.001604 diff --git a/runs/detect/train32/results.png b/runs/detect/train32/results.png new file mode 100644 index 0000000..ffe8af9 Binary files /dev/null and b/runs/detect/train32/results.png differ diff --git a/runs/detect/train32/train_batch0.jpg b/runs/detect/train32/train_batch0.jpg new file mode 100644 index 0000000..8f43221 Binary files /dev/null and b/runs/detect/train32/train_batch0.jpg differ diff --git a/runs/detect/train32/train_batch1.jpg b/runs/detect/train32/train_batch1.jpg new file mode 100644 index 0000000..c6ac35d Binary files /dev/null and b/runs/detect/train32/train_batch1.jpg differ diff --git a/runs/detect/train32/train_batch2.jpg b/runs/detect/train32/train_batch2.jpg new file mode 100644 index 0000000..8de88e2 Binary files /dev/null and b/runs/detect/train32/train_batch2.jpg differ diff --git a/runs/detect/train32/val_batch0_labels.jpg b/runs/detect/train32/val_batch0_labels.jpg new file mode 100644 index 0000000..78e0c13 Binary files /dev/null and b/runs/detect/train32/val_batch0_labels.jpg differ diff --git a/runs/detect/train32/val_batch0_pred.jpg b/runs/detect/train32/val_batch0_pred.jpg new file mode 100644 index 0000000..df21127 Binary files /dev/null and b/runs/detect/train32/val_batch0_pred.jpg differ diff --git a/runs/detect/train32/val_batch1_labels.jpg b/runs/detect/train32/val_batch1_labels.jpg new file mode 100644 index 0000000..e92d49d Binary files /dev/null and b/runs/detect/train32/val_batch1_labels.jpg differ diff --git a/runs/detect/train32/val_batch1_pred.jpg b/runs/detect/train32/val_batch1_pred.jpg new file mode 100644 index 0000000..f73c81f Binary files /dev/null and b/runs/detect/train32/val_batch1_pred.jpg differ diff --git a/runs/detect/train32/val_batch2_labels.jpg b/runs/detect/train32/val_batch2_labels.jpg new file mode 100644 index 0000000..fad7516 Binary files /dev/null and b/runs/detect/train32/val_batch2_labels.jpg differ diff --git a/runs/detect/train32/val_batch2_pred.jpg b/runs/detect/train32/val_batch2_pred.jpg new file mode 100644 index 0000000..a939841 Binary files /dev/null and b/runs/detect/train32/val_batch2_pred.jpg differ diff --git a/runs/detect/train32/weights/best.pt b/runs/detect/train32/weights/best.pt new file mode 100644 index 0000000..ea3da05 Binary files /dev/null and b/runs/detect/train32/weights/best.pt differ diff --git a/runs/detect/train32/weights/last.pt b/runs/detect/train32/weights/last.pt new file mode 100644 index 0000000..17159ac Binary files /dev/null and b/runs/detect/train32/weights/last.pt differ diff --git a/runs/detect/train322/F1_curve.png b/runs/detect/train322/F1_curve.png new file mode 100644 index 0000000..2fc8251 Binary files /dev/null and b/runs/detect/train322/F1_curve.png differ diff --git a/runs/detect/train322/PR_curve.png b/runs/detect/train322/PR_curve.png new file mode 100644 index 0000000..344050c Binary files /dev/null and b/runs/detect/train322/PR_curve.png differ diff --git a/runs/detect/train322/P_curve.png b/runs/detect/train322/P_curve.png new file mode 100644 index 0000000..a6b5a5e Binary files /dev/null and b/runs/detect/train322/P_curve.png differ diff --git a/runs/detect/train322/R_curve.png b/runs/detect/train322/R_curve.png new file mode 100644 index 0000000..2d44a85 Binary files /dev/null and b/runs/detect/train322/R_curve.png differ diff --git a/runs/detect/train322/confusion_matrix.png b/runs/detect/train322/confusion_matrix.png new file mode 100644 index 0000000..3d8ba2b Binary files /dev/null and b/runs/detect/train322/confusion_matrix.png differ diff --git a/runs/detect/train322/confusion_matrix_normalized.png b/runs/detect/train322/confusion_matrix_normalized.png new file mode 100644 index 0000000..ab149b8 Binary files /dev/null and b/runs/detect/train322/confusion_matrix_normalized.png differ diff --git a/runs/detect/train322/val_batch0_labels.jpg b/runs/detect/train322/val_batch0_labels.jpg new file mode 100644 index 0000000..6323831 Binary files /dev/null and b/runs/detect/train322/val_batch0_labels.jpg differ diff --git a/runs/detect/train322/val_batch0_pred.jpg b/runs/detect/train322/val_batch0_pred.jpg new file mode 100644 index 0000000..6a49803 Binary files /dev/null and b/runs/detect/train322/val_batch0_pred.jpg differ diff --git a/runs/detect/train322/val_batch1_labels.jpg b/runs/detect/train322/val_batch1_labels.jpg new file mode 100644 index 0000000..88a3ea2 Binary files /dev/null and b/runs/detect/train322/val_batch1_labels.jpg differ diff --git a/runs/detect/train322/val_batch1_pred.jpg b/runs/detect/train322/val_batch1_pred.jpg new file mode 100644 index 0000000..4a91dc4 Binary files /dev/null and b/runs/detect/train322/val_batch1_pred.jpg differ diff --git a/runs/detect/train322/val_batch2_labels.jpg b/runs/detect/train322/val_batch2_labels.jpg new file mode 100644 index 0000000..1443d38 Binary files /dev/null and b/runs/detect/train322/val_batch2_labels.jpg differ diff --git a/runs/detect/train322/val_batch2_pred.jpg b/runs/detect/train322/val_batch2_pred.jpg new file mode 100644 index 0000000..235e9ee Binary files /dev/null and b/runs/detect/train322/val_batch2_pred.jpg differ diff --git a/src/data.py b/src/data.py index 3dfab1c..e7d451d 100644 --- a/src/data.py +++ b/src/data.py @@ -4,24 +4,13 @@ import os from PIL import Image,ImageEnhance from PIL import Image import os - -# 指定文件夹路径 -folder_path = 'your_folder_path' - -# 遍历文件夹中的所有文件 +folder_path = 'C:/workspace/le-yolo/data/images/train' for filename in os.listdir(folder_path): - # 检查文件是否是图片(这里以常见的图片格式为例) if filename.lower().endswith(('.png', '.jpg', '.jpeg', '.bmp', '.gif')): - # 打开图片 img_path = os.path.join(folder_path, filename) img = Image.open(img_path) - - # 转换为灰度图像 gray_img = img.convert('L') - - # 保存灰度图像(可以保存到原文件夹或指定的新文件夹) - # 这里以在原文件夹保存为例,文件名不变,只是修改了内容 - gray_img.save(img_path) - - print(f"已将 {filename} 转换为灰度图像并保存") - + new_filename = os.path.splitext(filename)[0] + '_gray.jpg' # 修改为新文件名,如 "image_gray.jpg" + new_img_path = os.path.join(folder_path, new_filename) + gray_img.save(new_img_path) + print(f"已将 {filename} 转换为灰度图像并保存") \ No newline at end of file diff --git a/src/data.yaml b/src/data.yaml index 3cabfd0..7b1c8a9 100644 --- a/src/data.yaml +++ b/src/data.yaml @@ -1,6 +1,6 @@ path: C:\workspace\le-yolo\data -train: images/train -val: images/test -test: images/val +train: images/test +val: images/val +test: images/train nc: 1 names: [ 'person' ] \ No newline at end of file diff --git a/src/detected_objects/detected_object_0_1.0.jpg b/src/detected_objects/detected_object_0_1.0.jpg new file mode 100644 index 0000000..1fbe204 Binary files /dev/null and b/src/detected_objects/detected_object_0_1.0.jpg differ diff --git a/src/detected_objects/detected_object_0_10.0.jpg b/src/detected_objects/detected_object_0_10.0.jpg new file mode 100644 index 0000000..d44c8cf Binary files /dev/null and b/src/detected_objects/detected_object_0_10.0.jpg differ diff --git a/src/detected_objects/detected_object_0_100.0.jpg b/src/detected_objects/detected_object_0_100.0.jpg new file mode 100644 index 0000000..1cda9ca Binary files /dev/null and b/src/detected_objects/detected_object_0_100.0.jpg differ diff --git a/src/detected_objects/detected_object_0_101.0.jpg b/src/detected_objects/detected_object_0_101.0.jpg new file mode 100644 index 0000000..8d27e88 Binary files /dev/null and b/src/detected_objects/detected_object_0_101.0.jpg differ diff --git a/src/detected_objects/detected_object_0_102.0.jpg b/src/detected_objects/detected_object_0_102.0.jpg new file mode 100644 index 0000000..6411a9a Binary files /dev/null and b/src/detected_objects/detected_object_0_102.0.jpg differ diff --git a/src/detected_objects/detected_object_0_103.0.jpg b/src/detected_objects/detected_object_0_103.0.jpg new file mode 100644 index 0000000..363fb3d Binary files /dev/null and b/src/detected_objects/detected_object_0_103.0.jpg differ diff --git a/src/detected_objects/detected_object_0_104.0.jpg b/src/detected_objects/detected_object_0_104.0.jpg new file mode 100644 index 0000000..bc96cb2 Binary files /dev/null and b/src/detected_objects/detected_object_0_104.0.jpg differ diff --git a/src/detected_objects/detected_object_0_105.0.jpg b/src/detected_objects/detected_object_0_105.0.jpg new file mode 100644 index 0000000..1d7507a Binary files /dev/null and b/src/detected_objects/detected_object_0_105.0.jpg differ diff --git a/src/detected_objects/detected_object_0_106.0.jpg b/src/detected_objects/detected_object_0_106.0.jpg new file mode 100644 index 0000000..f5dce91 Binary files /dev/null and b/src/detected_objects/detected_object_0_106.0.jpg differ diff --git a/src/detected_objects/detected_object_0_107.0.jpg b/src/detected_objects/detected_object_0_107.0.jpg new file mode 100644 index 0000000..dbb43fe Binary files /dev/null and b/src/detected_objects/detected_object_0_107.0.jpg differ diff --git a/src/detected_objects/detected_object_0_108.0.jpg b/src/detected_objects/detected_object_0_108.0.jpg new file mode 100644 index 0000000..e3493b5 Binary files /dev/null and b/src/detected_objects/detected_object_0_108.0.jpg differ diff --git a/src/detected_objects/detected_object_0_109.0.jpg b/src/detected_objects/detected_object_0_109.0.jpg new file mode 100644 index 0000000..5272488 Binary files /dev/null and b/src/detected_objects/detected_object_0_109.0.jpg differ diff --git a/src/detected_objects/detected_object_0_11.0.jpg b/src/detected_objects/detected_object_0_11.0.jpg new file mode 100644 index 0000000..c692fce Binary files /dev/null and b/src/detected_objects/detected_object_0_11.0.jpg differ diff --git a/src/detected_objects/detected_object_0_110.0.jpg b/src/detected_objects/detected_object_0_110.0.jpg new file mode 100644 index 0000000..5c737d9 Binary files /dev/null and b/src/detected_objects/detected_object_0_110.0.jpg differ diff --git a/src/detected_objects/detected_object_0_111.0.jpg b/src/detected_objects/detected_object_0_111.0.jpg new file mode 100644 index 0000000..f140724 Binary files /dev/null and b/src/detected_objects/detected_object_0_111.0.jpg differ diff --git a/src/detected_objects/detected_object_0_112.0.jpg b/src/detected_objects/detected_object_0_112.0.jpg new file mode 100644 index 0000000..7519663 Binary files /dev/null and b/src/detected_objects/detected_object_0_112.0.jpg differ diff --git a/src/detected_objects/detected_object_0_113.0.jpg b/src/detected_objects/detected_object_0_113.0.jpg new file mode 100644 index 0000000..30a9249 Binary files /dev/null and b/src/detected_objects/detected_object_0_113.0.jpg differ diff --git a/src/detected_objects/detected_object_0_114.0.jpg b/src/detected_objects/detected_object_0_114.0.jpg new file mode 100644 index 0000000..fb17e0c Binary files /dev/null and b/src/detected_objects/detected_object_0_114.0.jpg differ diff --git a/src/detected_objects/detected_object_0_115.0.jpg b/src/detected_objects/detected_object_0_115.0.jpg new file mode 100644 index 0000000..837ecb1 Binary files /dev/null and b/src/detected_objects/detected_object_0_115.0.jpg differ diff --git a/src/detected_objects/detected_object_0_12.0.jpg b/src/detected_objects/detected_object_0_12.0.jpg new file mode 100644 index 0000000..3724b44 Binary files /dev/null and b/src/detected_objects/detected_object_0_12.0.jpg differ diff --git a/src/detected_objects/detected_object_0_13.0.jpg b/src/detected_objects/detected_object_0_13.0.jpg new file mode 100644 index 0000000..3dbdbfd Binary files /dev/null and b/src/detected_objects/detected_object_0_13.0.jpg differ diff --git a/src/detected_objects/detected_object_0_14.0.jpg b/src/detected_objects/detected_object_0_14.0.jpg new file mode 100644 index 0000000..9f76d74 Binary files /dev/null and b/src/detected_objects/detected_object_0_14.0.jpg differ diff --git a/src/detected_objects/detected_object_0_15.0.jpg b/src/detected_objects/detected_object_0_15.0.jpg new file mode 100644 index 0000000..b998266 Binary files /dev/null and b/src/detected_objects/detected_object_0_15.0.jpg differ diff --git a/src/detected_objects/detected_object_0_16.0.jpg b/src/detected_objects/detected_object_0_16.0.jpg new file mode 100644 index 0000000..d952666 Binary files /dev/null and b/src/detected_objects/detected_object_0_16.0.jpg differ diff --git a/src/detected_objects/detected_object_0_17.0.jpg b/src/detected_objects/detected_object_0_17.0.jpg new file mode 100644 index 0000000..f623a4b Binary files /dev/null and b/src/detected_objects/detected_object_0_17.0.jpg differ diff --git a/src/detected_objects/detected_object_0_18.0.jpg b/src/detected_objects/detected_object_0_18.0.jpg new file mode 100644 index 0000000..9de5566 Binary files /dev/null and b/src/detected_objects/detected_object_0_18.0.jpg differ diff --git a/src/detected_objects/detected_object_0_19.0.jpg b/src/detected_objects/detected_object_0_19.0.jpg new file mode 100644 index 0000000..ff2ff33 Binary files /dev/null and b/src/detected_objects/detected_object_0_19.0.jpg differ diff --git a/src/detected_objects/detected_object_0_2.0.jpg b/src/detected_objects/detected_object_0_2.0.jpg new file mode 100644 index 0000000..425a22a Binary files /dev/null and b/src/detected_objects/detected_object_0_2.0.jpg differ diff --git a/src/detected_objects/detected_object_0_20.0.jpg b/src/detected_objects/detected_object_0_20.0.jpg new file mode 100644 index 0000000..835f480 Binary files /dev/null and b/src/detected_objects/detected_object_0_20.0.jpg differ diff --git a/src/detected_objects/detected_object_0_21.0.jpg b/src/detected_objects/detected_object_0_21.0.jpg new file mode 100644 index 0000000..8af5e21 Binary files /dev/null and b/src/detected_objects/detected_object_0_21.0.jpg differ diff --git a/src/detected_objects/detected_object_0_22.0.jpg b/src/detected_objects/detected_object_0_22.0.jpg new file mode 100644 index 0000000..4b6db1c Binary files /dev/null and b/src/detected_objects/detected_object_0_22.0.jpg differ diff --git a/src/detected_objects/detected_object_0_23.0.jpg b/src/detected_objects/detected_object_0_23.0.jpg new file mode 100644 index 0000000..2136cc2 Binary files /dev/null and b/src/detected_objects/detected_object_0_23.0.jpg differ diff --git a/src/detected_objects/detected_object_0_24.0.jpg b/src/detected_objects/detected_object_0_24.0.jpg new file mode 100644 index 0000000..102d457 Binary files /dev/null and b/src/detected_objects/detected_object_0_24.0.jpg differ diff --git a/src/detected_objects/detected_object_0_25.0.jpg b/src/detected_objects/detected_object_0_25.0.jpg new file mode 100644 index 0000000..9b69d32 Binary files /dev/null and b/src/detected_objects/detected_object_0_25.0.jpg differ diff --git a/src/detected_objects/detected_object_0_26.0.jpg b/src/detected_objects/detected_object_0_26.0.jpg new file mode 100644 index 0000000..fbef6fe Binary files /dev/null and b/src/detected_objects/detected_object_0_26.0.jpg differ diff --git a/src/detected_objects/detected_object_0_27.0.jpg b/src/detected_objects/detected_object_0_27.0.jpg new file mode 100644 index 0000000..a6c481a Binary files /dev/null and b/src/detected_objects/detected_object_0_27.0.jpg differ diff --git a/src/detected_objects/detected_object_0_28.0.jpg b/src/detected_objects/detected_object_0_28.0.jpg new file mode 100644 index 0000000..fc84e9b Binary files /dev/null and b/src/detected_objects/detected_object_0_28.0.jpg differ diff --git a/src/detected_objects/detected_object_0_29.0.jpg b/src/detected_objects/detected_object_0_29.0.jpg new file mode 100644 index 0000000..0d5b6d4 Binary files /dev/null and b/src/detected_objects/detected_object_0_29.0.jpg differ diff --git a/src/detected_objects/detected_object_0_3.0.jpg b/src/detected_objects/detected_object_0_3.0.jpg new file mode 100644 index 0000000..fc438c3 Binary files /dev/null and b/src/detected_objects/detected_object_0_3.0.jpg differ diff --git a/src/detected_objects/detected_object_0_30.0.jpg b/src/detected_objects/detected_object_0_30.0.jpg new file mode 100644 index 0000000..2881be5 Binary files /dev/null and b/src/detected_objects/detected_object_0_30.0.jpg differ diff --git a/src/detected_objects/detected_object_0_31.0.jpg b/src/detected_objects/detected_object_0_31.0.jpg new file mode 100644 index 0000000..4805185 Binary files /dev/null and b/src/detected_objects/detected_object_0_31.0.jpg differ diff --git a/src/detected_objects/detected_object_0_32.0.jpg b/src/detected_objects/detected_object_0_32.0.jpg new file mode 100644 index 0000000..ce56e9c Binary files /dev/null and b/src/detected_objects/detected_object_0_32.0.jpg differ diff --git a/src/detected_objects/detected_object_0_33.0.jpg b/src/detected_objects/detected_object_0_33.0.jpg new file mode 100644 index 0000000..40038ce Binary files /dev/null and b/src/detected_objects/detected_object_0_33.0.jpg differ diff --git a/src/detected_objects/detected_object_0_34.0.jpg b/src/detected_objects/detected_object_0_34.0.jpg new file mode 100644 index 0000000..2615a64 Binary files /dev/null and b/src/detected_objects/detected_object_0_34.0.jpg differ diff --git a/src/detected_objects/detected_object_0_35.0.jpg b/src/detected_objects/detected_object_0_35.0.jpg new file mode 100644 index 0000000..2b559a6 Binary files /dev/null and b/src/detected_objects/detected_object_0_35.0.jpg differ diff --git a/src/detected_objects/detected_object_0_36.0.jpg b/src/detected_objects/detected_object_0_36.0.jpg new file mode 100644 index 0000000..d44c85d Binary files /dev/null and b/src/detected_objects/detected_object_0_36.0.jpg differ diff --git a/src/detected_objects/detected_object_0_37.0.jpg b/src/detected_objects/detected_object_0_37.0.jpg new file mode 100644 index 0000000..1d4d0fa Binary files /dev/null and b/src/detected_objects/detected_object_0_37.0.jpg differ diff --git a/src/detected_objects/detected_object_0_38.0.jpg b/src/detected_objects/detected_object_0_38.0.jpg new file mode 100644 index 0000000..02ffd84 Binary files /dev/null and b/src/detected_objects/detected_object_0_38.0.jpg differ diff --git a/src/detected_objects/detected_object_0_39.0.jpg b/src/detected_objects/detected_object_0_39.0.jpg new file mode 100644 index 0000000..69f1b93 Binary files /dev/null and b/src/detected_objects/detected_object_0_39.0.jpg differ diff --git a/src/detected_objects/detected_object_0_4.0.jpg b/src/detected_objects/detected_object_0_4.0.jpg new file mode 100644 index 0000000..b73f75f Binary files /dev/null and b/src/detected_objects/detected_object_0_4.0.jpg differ diff --git a/src/detected_objects/detected_object_0_40.0.jpg b/src/detected_objects/detected_object_0_40.0.jpg new file mode 100644 index 0000000..3337ddc Binary files /dev/null and b/src/detected_objects/detected_object_0_40.0.jpg differ diff --git a/src/detected_objects/detected_object_0_41.0.jpg b/src/detected_objects/detected_object_0_41.0.jpg new file mode 100644 index 0000000..46a4833 Binary files /dev/null and b/src/detected_objects/detected_object_0_41.0.jpg differ diff --git a/src/detected_objects/detected_object_0_42.0.jpg b/src/detected_objects/detected_object_0_42.0.jpg new file mode 100644 index 0000000..4355b69 Binary files /dev/null and b/src/detected_objects/detected_object_0_42.0.jpg differ diff --git a/src/detected_objects/detected_object_0_43.0.jpg b/src/detected_objects/detected_object_0_43.0.jpg new file mode 100644 index 0000000..f5f247c Binary files /dev/null and b/src/detected_objects/detected_object_0_43.0.jpg differ diff --git a/src/detected_objects/detected_object_0_44.0.jpg b/src/detected_objects/detected_object_0_44.0.jpg new file mode 100644 index 0000000..aca754b Binary files /dev/null and b/src/detected_objects/detected_object_0_44.0.jpg differ diff --git a/src/detected_objects/detected_object_0_45.0.jpg b/src/detected_objects/detected_object_0_45.0.jpg new file mode 100644 index 0000000..b5287c3 Binary files /dev/null and b/src/detected_objects/detected_object_0_45.0.jpg differ diff --git a/src/detected_objects/detected_object_0_46.0.jpg b/src/detected_objects/detected_object_0_46.0.jpg new file mode 100644 index 0000000..1613041 Binary files /dev/null and b/src/detected_objects/detected_object_0_46.0.jpg differ diff --git a/src/detected_objects/detected_object_0_47.0.jpg b/src/detected_objects/detected_object_0_47.0.jpg new file mode 100644 index 0000000..76d089e Binary files /dev/null and b/src/detected_objects/detected_object_0_47.0.jpg differ diff --git a/src/detected_objects/detected_object_0_48.0.jpg b/src/detected_objects/detected_object_0_48.0.jpg new file mode 100644 index 0000000..d0adb03 Binary files /dev/null and b/src/detected_objects/detected_object_0_48.0.jpg differ diff --git a/src/detected_objects/detected_object_0_49.0.jpg b/src/detected_objects/detected_object_0_49.0.jpg new file mode 100644 index 0000000..39275f7 Binary files /dev/null and b/src/detected_objects/detected_object_0_49.0.jpg differ diff --git a/src/detected_objects/detected_object_0_5.0.jpg b/src/detected_objects/detected_object_0_5.0.jpg new file mode 100644 index 0000000..e6a5e1f Binary files /dev/null and b/src/detected_objects/detected_object_0_5.0.jpg differ diff --git a/src/detected_objects/detected_object_0_50.0.jpg b/src/detected_objects/detected_object_0_50.0.jpg new file mode 100644 index 0000000..c651993 Binary files /dev/null and b/src/detected_objects/detected_object_0_50.0.jpg differ diff --git a/src/detected_objects/detected_object_0_51.0.jpg b/src/detected_objects/detected_object_0_51.0.jpg new file mode 100644 index 0000000..0d31dbc Binary files /dev/null and b/src/detected_objects/detected_object_0_51.0.jpg differ diff --git a/src/detected_objects/detected_object_0_52.0.jpg b/src/detected_objects/detected_object_0_52.0.jpg new file mode 100644 index 0000000..d2245e8 Binary files /dev/null and b/src/detected_objects/detected_object_0_52.0.jpg differ diff --git a/src/detected_objects/detected_object_0_53.0.jpg b/src/detected_objects/detected_object_0_53.0.jpg new file mode 100644 index 0000000..0c62789 Binary files /dev/null and b/src/detected_objects/detected_object_0_53.0.jpg differ diff --git a/src/detected_objects/detected_object_0_54.0.jpg b/src/detected_objects/detected_object_0_54.0.jpg new file mode 100644 index 0000000..8dc54de Binary files /dev/null and b/src/detected_objects/detected_object_0_54.0.jpg differ diff --git a/src/detected_objects/detected_object_0_55.0.jpg b/src/detected_objects/detected_object_0_55.0.jpg new file mode 100644 index 0000000..33223ca Binary files /dev/null and b/src/detected_objects/detected_object_0_55.0.jpg differ diff --git a/src/detected_objects/detected_object_0_56.0.jpg b/src/detected_objects/detected_object_0_56.0.jpg new file mode 100644 index 0000000..f4d46bc Binary files /dev/null and b/src/detected_objects/detected_object_0_56.0.jpg differ diff --git a/src/detected_objects/detected_object_0_57.0.jpg b/src/detected_objects/detected_object_0_57.0.jpg new file mode 100644 index 0000000..35e4402 Binary files /dev/null and b/src/detected_objects/detected_object_0_57.0.jpg differ diff --git a/src/detected_objects/detected_object_0_58.0.jpg b/src/detected_objects/detected_object_0_58.0.jpg new file mode 100644 index 0000000..1b3a823 Binary files /dev/null and b/src/detected_objects/detected_object_0_58.0.jpg differ diff --git a/src/detected_objects/detected_object_0_59.0.jpg b/src/detected_objects/detected_object_0_59.0.jpg new file mode 100644 index 0000000..b5e7dae Binary files /dev/null and b/src/detected_objects/detected_object_0_59.0.jpg differ diff --git a/src/detected_objects/detected_object_0_6.0.jpg b/src/detected_objects/detected_object_0_6.0.jpg new file mode 100644 index 0000000..9410eeb Binary files /dev/null and b/src/detected_objects/detected_object_0_6.0.jpg differ diff --git a/src/detected_objects/detected_object_0_60.0.jpg b/src/detected_objects/detected_object_0_60.0.jpg new file mode 100644 index 0000000..2529bc2 Binary files /dev/null and b/src/detected_objects/detected_object_0_60.0.jpg differ diff --git a/src/detected_objects/detected_object_0_61.0.jpg b/src/detected_objects/detected_object_0_61.0.jpg new file mode 100644 index 0000000..c4ad76e Binary files /dev/null and b/src/detected_objects/detected_object_0_61.0.jpg differ diff --git a/src/detected_objects/detected_object_0_62.0.jpg b/src/detected_objects/detected_object_0_62.0.jpg new file mode 100644 index 0000000..08e4127 Binary files /dev/null and b/src/detected_objects/detected_object_0_62.0.jpg differ diff --git a/src/detected_objects/detected_object_0_63.0.jpg b/src/detected_objects/detected_object_0_63.0.jpg new file mode 100644 index 0000000..8d67ffe Binary files /dev/null and b/src/detected_objects/detected_object_0_63.0.jpg differ diff --git a/src/detected_objects/detected_object_0_64.0.jpg b/src/detected_objects/detected_object_0_64.0.jpg new file mode 100644 index 0000000..9685fd5 Binary files /dev/null and b/src/detected_objects/detected_object_0_64.0.jpg differ diff --git a/src/detected_objects/detected_object_0_65.0.jpg b/src/detected_objects/detected_object_0_65.0.jpg new file mode 100644 index 0000000..c44e33e Binary files /dev/null and b/src/detected_objects/detected_object_0_65.0.jpg differ diff --git a/src/detected_objects/detected_object_0_66.0.jpg b/src/detected_objects/detected_object_0_66.0.jpg new file mode 100644 index 0000000..67c7a3b Binary files /dev/null and b/src/detected_objects/detected_object_0_66.0.jpg differ diff --git a/src/detected_objects/detected_object_0_67.0.jpg b/src/detected_objects/detected_object_0_67.0.jpg new file mode 100644 index 0000000..d5f60c9 Binary files /dev/null and b/src/detected_objects/detected_object_0_67.0.jpg differ diff --git a/src/detected_objects/detected_object_0_68.0.jpg b/src/detected_objects/detected_object_0_68.0.jpg new file mode 100644 index 0000000..7187cfe Binary files /dev/null and b/src/detected_objects/detected_object_0_68.0.jpg differ diff --git a/src/detected_objects/detected_object_0_69.0.jpg b/src/detected_objects/detected_object_0_69.0.jpg new file mode 100644 index 0000000..d9d9222 Binary files /dev/null and b/src/detected_objects/detected_object_0_69.0.jpg differ diff --git a/src/detected_objects/detected_object_0_7.0.jpg b/src/detected_objects/detected_object_0_7.0.jpg new file mode 100644 index 0000000..d04ba3b Binary files /dev/null and b/src/detected_objects/detected_object_0_7.0.jpg differ diff --git a/src/detected_objects/detected_object_0_70.0.jpg b/src/detected_objects/detected_object_0_70.0.jpg new file mode 100644 index 0000000..e5aed7a Binary files /dev/null and b/src/detected_objects/detected_object_0_70.0.jpg differ diff --git a/src/detected_objects/detected_object_0_71.0.jpg b/src/detected_objects/detected_object_0_71.0.jpg new file mode 100644 index 0000000..d13cb10 Binary files /dev/null and b/src/detected_objects/detected_object_0_71.0.jpg differ diff --git a/src/detected_objects/detected_object_0_72.0.jpg b/src/detected_objects/detected_object_0_72.0.jpg new file mode 100644 index 0000000..330a848 Binary files /dev/null and b/src/detected_objects/detected_object_0_72.0.jpg differ diff --git a/src/detected_objects/detected_object_0_73.0.jpg b/src/detected_objects/detected_object_0_73.0.jpg new file mode 100644 index 0000000..b7a742b Binary files /dev/null and b/src/detected_objects/detected_object_0_73.0.jpg differ diff --git a/src/detected_objects/detected_object_0_74.0.jpg b/src/detected_objects/detected_object_0_74.0.jpg new file mode 100644 index 0000000..1ab3185 Binary files /dev/null and b/src/detected_objects/detected_object_0_74.0.jpg differ diff --git a/src/detected_objects/detected_object_0_75.0.jpg b/src/detected_objects/detected_object_0_75.0.jpg new file mode 100644 index 0000000..a3ebd18 Binary files /dev/null and b/src/detected_objects/detected_object_0_75.0.jpg differ diff --git a/src/detected_objects/detected_object_0_76.0.jpg b/src/detected_objects/detected_object_0_76.0.jpg new file mode 100644 index 0000000..9c95cf8 Binary files /dev/null and b/src/detected_objects/detected_object_0_76.0.jpg differ diff --git a/src/detected_objects/detected_object_0_77.0.jpg b/src/detected_objects/detected_object_0_77.0.jpg new file mode 100644 index 0000000..5a33d2d Binary files /dev/null and b/src/detected_objects/detected_object_0_77.0.jpg differ diff --git a/src/detected_objects/detected_object_0_78.0.jpg b/src/detected_objects/detected_object_0_78.0.jpg new file mode 100644 index 0000000..ea82dac Binary files /dev/null and b/src/detected_objects/detected_object_0_78.0.jpg differ diff --git a/src/detected_objects/detected_object_0_79.0.jpg b/src/detected_objects/detected_object_0_79.0.jpg new file mode 100644 index 0000000..5831ab2 Binary files /dev/null and b/src/detected_objects/detected_object_0_79.0.jpg differ diff --git a/src/detected_objects/detected_object_0_8.0.jpg b/src/detected_objects/detected_object_0_8.0.jpg new file mode 100644 index 0000000..f6f5633 Binary files /dev/null and b/src/detected_objects/detected_object_0_8.0.jpg differ diff --git a/src/detected_objects/detected_object_0_80.0.jpg b/src/detected_objects/detected_object_0_80.0.jpg new file mode 100644 index 0000000..c7dbaa0 Binary files /dev/null and b/src/detected_objects/detected_object_0_80.0.jpg differ diff --git a/src/detected_objects/detected_object_0_81.0.jpg b/src/detected_objects/detected_object_0_81.0.jpg new file mode 100644 index 0000000..ab7488f Binary files /dev/null and b/src/detected_objects/detected_object_0_81.0.jpg differ diff --git a/src/detected_objects/detected_object_0_82.0.jpg b/src/detected_objects/detected_object_0_82.0.jpg new file mode 100644 index 0000000..f0b7a8d Binary files /dev/null and b/src/detected_objects/detected_object_0_82.0.jpg differ diff --git a/src/detected_objects/detected_object_0_83.0.jpg b/src/detected_objects/detected_object_0_83.0.jpg new file mode 100644 index 0000000..4bf72e5 Binary files /dev/null and b/src/detected_objects/detected_object_0_83.0.jpg differ diff --git a/src/detected_objects/detected_object_0_84.0.jpg b/src/detected_objects/detected_object_0_84.0.jpg new file mode 100644 index 0000000..f5f58d1 Binary files /dev/null and b/src/detected_objects/detected_object_0_84.0.jpg differ diff --git a/src/detected_objects/detected_object_0_85.0.jpg b/src/detected_objects/detected_object_0_85.0.jpg new file mode 100644 index 0000000..173c2ed Binary files /dev/null and b/src/detected_objects/detected_object_0_85.0.jpg differ diff --git a/src/detected_objects/detected_object_0_86.0.jpg b/src/detected_objects/detected_object_0_86.0.jpg new file mode 100644 index 0000000..a0385fd Binary files /dev/null and b/src/detected_objects/detected_object_0_86.0.jpg differ diff --git a/src/detected_objects/detected_object_0_87.0.jpg b/src/detected_objects/detected_object_0_87.0.jpg new file mode 100644 index 0000000..ba1e648 Binary files /dev/null and b/src/detected_objects/detected_object_0_87.0.jpg differ diff --git a/src/detected_objects/detected_object_0_88.0.jpg b/src/detected_objects/detected_object_0_88.0.jpg new file mode 100644 index 0000000..25585e2 Binary files /dev/null and b/src/detected_objects/detected_object_0_88.0.jpg differ diff --git a/src/detected_objects/detected_object_0_89.0.jpg b/src/detected_objects/detected_object_0_89.0.jpg new file mode 100644 index 0000000..7995ab3 Binary files /dev/null and b/src/detected_objects/detected_object_0_89.0.jpg differ diff --git a/src/detected_objects/detected_object_0_9.0.jpg b/src/detected_objects/detected_object_0_9.0.jpg new file mode 100644 index 0000000..95121f6 Binary files /dev/null and b/src/detected_objects/detected_object_0_9.0.jpg differ diff --git a/src/detected_objects/detected_object_0_90.0.jpg b/src/detected_objects/detected_object_0_90.0.jpg new file mode 100644 index 0000000..ed01f25 Binary files /dev/null and b/src/detected_objects/detected_object_0_90.0.jpg differ diff --git a/src/detected_objects/detected_object_0_91.0.jpg b/src/detected_objects/detected_object_0_91.0.jpg new file mode 100644 index 0000000..02643b5 Binary files /dev/null and b/src/detected_objects/detected_object_0_91.0.jpg differ diff --git a/src/detected_objects/detected_object_0_92.0.jpg b/src/detected_objects/detected_object_0_92.0.jpg new file mode 100644 index 0000000..eb8ca3d Binary files /dev/null and b/src/detected_objects/detected_object_0_92.0.jpg differ diff --git a/src/detected_objects/detected_object_0_93.0.jpg b/src/detected_objects/detected_object_0_93.0.jpg new file mode 100644 index 0000000..ca2b93c Binary files /dev/null and b/src/detected_objects/detected_object_0_93.0.jpg differ diff --git a/src/detected_objects/detected_object_0_94.0.jpg b/src/detected_objects/detected_object_0_94.0.jpg new file mode 100644 index 0000000..a58a53a Binary files /dev/null and b/src/detected_objects/detected_object_0_94.0.jpg differ diff --git a/src/detected_objects/detected_object_0_95.0.jpg b/src/detected_objects/detected_object_0_95.0.jpg new file mode 100644 index 0000000..e1cf57c Binary files /dev/null and b/src/detected_objects/detected_object_0_95.0.jpg differ diff --git a/src/detected_objects/detected_object_0_96.0.jpg b/src/detected_objects/detected_object_0_96.0.jpg new file mode 100644 index 0000000..58f0d6b Binary files /dev/null and b/src/detected_objects/detected_object_0_96.0.jpg differ diff --git a/src/detected_objects/detected_object_0_97.0.jpg b/src/detected_objects/detected_object_0_97.0.jpg new file mode 100644 index 0000000..9ecb706 Binary files /dev/null and b/src/detected_objects/detected_object_0_97.0.jpg differ diff --git a/src/detected_objects/detected_object_0_98.0.jpg b/src/detected_objects/detected_object_0_98.0.jpg new file mode 100644 index 0000000..d14482b Binary files /dev/null and b/src/detected_objects/detected_object_0_98.0.jpg differ diff --git a/src/detected_objects/detected_object_0_99.0.jpg b/src/detected_objects/detected_object_0_99.0.jpg new file mode 100644 index 0000000..2fec66a Binary files /dev/null and b/src/detected_objects/detected_object_0_99.0.jpg differ diff --git a/src/pic.py b/src/pic.py index 8f65343..bb2f44f 100644 --- a/src/pic.py +++ b/src/pic.py @@ -1,88 +1,29 @@ -import os +from ultralytics import YOLO import cv2 -def get_class_folder(catagetory, class_id, base_folder): - # 根据类别ID创建文件夹路径 - class_folder = os.path.join(base_folder, catagetory[int(class_id)]) - if not os.path.exists(class_folder): - os.makedirs(class_folder) - return class_folder - -def crop_and_save(image_path, label, class_folder): - # 读取图片 - img = cv2.imread(image_path) - height, width = img.shape[:2] - - # 解析标签 - parts = label.strip().split() - - x_center = float(parts[1]) * width - y_center = float(parts[2]) * height - img_width = float(parts[3]) * width - img_height = float(parts[4]) * height - - # 计算左上角坐标 - x1 = int(x_center - (img_width / 2)) - y1 = int(y_center - (img_height / 2)) - - # 确保裁剪区域在图片范围内 - x1 = max(0, x1) - y1 = max(0, y1) - x2 = min(width, x1 + img_width) - y2 = min(height, y1 + img_height) - - # 裁剪图片 - cropped_img = img[int(y1):int(y2), int(x1):int(x2)] - - # 保存裁剪后的图片,文件名为数字递增 - file_number = 1 - file_path = os.path.join(class_folder, f"{file_number}.jpg") - while os.path.exists(file_path): - file_number += 1 - file_path = os.path.join(class_folder, f"{file_number}.jpg") - - cv2.imwrite(file_path, cropped_img) - - -def main(): - root = r"D:deskAdd_Building_20240616IR" - images_folder = os.path.join(root, 'images') - labels_folder = os.path.join(root, 'labels') - cropped_images_base_folder = os.path.join(root, 'cropped_images') - class_path = os.path.join(labels_folder, r"classes.txt") - category = {} - with open(class_path, 'r', encoding='utf-8') as f: - cls = f.readlines() - for i, c in enumerate(cls): - category[i] = c.strip() - - # 确保裁剪图片的基文件夹存在 - if not os.path.exists(cropped_images_base_folder): - os.makedirs(cropped_images_base_folder) - - # 遍历images文件夹中的所有图片 - for image_filename in os.listdir(images_folder): - if image_filename.lower().endswith(('.png', '.jpg', '.jpeg')): - # 构建图片和标签的完整路径 - image_path = os.path.join(images_folder, image_filename) - label_path = os.path.join(labels_folder, os.path.splitext(image_filename)[0] + '.txt') - - # 确保标签文件存在 - if os.path.exists(label_path): - # 读取标签文件 - with open(label_path, 'r') as file: - labels = file.readlines() - - # 遍历每个标签 - for label in labels: - # 获取类别ID - class_id = int(label.strip().split()[0]) - - # 获取类别文件夹 - class_folder = get_class_folder(category, class_id, cropped_images_base_folder) - - # 裁剪并保存图片 - crop_and_save(image_path, label, class_folder) - - -if __name__ == '__main__': - main() \ No newline at end of file +import os +model = YOLO(r"C:\workspace\le-yolo\runs\detect\train32\weights\best.pt") +target_class = 0 +cap = cv2.VideoCapture('../res/1.mp4') +if not os.path.exists('detected_objects'): + os.makedirs('detected_objects') +while cap.isOpened(): + success, frame = cap.read() + if not success: + break + results = model.predict(frame) + for result in results: + boxes = result.boxes + for box in boxes: + cls_id = box.cls.item() + if cls_id == target_class: + x1, y1, x2, y2 = map(int, box.xyxy[0].tolist()) + detected_object = frame[y1:y2, x1:x2] + save_path = os.path.join('detected_objects', f'detected_object_{target_class}_{cap.get(cv2.CAP_PROP_POS_FRAMES)}.jpg') + cv2.imwrite(save_path, detected_object) + print(f"截图保存至:{save_path}") + annotated_frame = results[0].plot() + cv2.imshow('YOLOv8 Detection', annotated_frame) + if cv2.waitKey(1) & 0xFF == ord(' '): + break +cap.release() +cv2.destroyAllWindows() \ No newline at end of file diff --git a/src/test.py b/src/test.py index b1a14f1..1875b89 100644 --- a/src/test.py +++ b/src/test.py @@ -1,3 +1,3 @@ from ultralytics import YOLO -model = YOLO(r"C:\workspace\le-yolo\runs\detect\train28\weights\best.pt") -results = model.predict("../res/4.mp4", show=True, save=True) +model = YOLO(r"C:\workspace\le-yolo\runs\detect\train32\weights\best.pt") +results = model.predict("../res/3.mp4", show=True, save=True) diff --git a/src/train.py b/src/train.py index c7abe9b..e52ae7c 100644 --- a/src/train.py +++ b/src/train.py @@ -1,5 +1,5 @@ from ultralytics import YOLO -model = YOLO(r"C:\workspace\le-yolo\runs\detect\train28\weights\last.pt") -model.train(data="data.yaml", epochs=500, batch=8, device='cpu', imgsz=640) +model = YOLO(r"C:\workspace\le-yolo\runs\detect\train32\weights\last.pt") +model.train(data="data.yaml", epochs=100, batch=8, device='cpu', imgsz=640) model.val() print('训练完成') \ No newline at end of file