Add trained YOLOv8 PPE detection model

.gitattributes

@@ -33,3 +33,27 @@ saved_model/**/* filter=lfs diff=lfs merge=lfs -text
 *.zip filter=lfs diff=lfs merge=lfs -text
 *.zst filter=lfs diff=lfs merge=lfs -text
 *tfevents* filter=lfs diff=lfs merge=lfs -text
+BoxF1_curve.png filter=lfs diff=lfs merge=lfs -text
+BoxPR_curve.png filter=lfs diff=lfs merge=lfs -text
+BoxP_curve.png filter=lfs diff=lfs merge=lfs -text
+BoxR_curve.png filter=lfs diff=lfs merge=lfs -text
+confusion_matrix.png filter=lfs diff=lfs merge=lfs -text
+confusion_matrix_normalized.png filter=lfs diff=lfs merge=lfs -text
+labels.jpg filter=lfs diff=lfs merge=lfs -text
+results.png filter=lfs diff=lfs merge=lfs -text
+train_batch0.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch1.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch2.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch7020.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch7021.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch7022.jpg filter=lfs diff=lfs merge=lfs -text
+training_curves.png filter=lfs diff=lfs merge=lfs -text
+val_batch0_labels.jpg filter=lfs diff=lfs merge=lfs -text
+val_batch0_pred.jpg filter=lfs diff=lfs merge=lfs -text
+val_batch1_labels.jpg filter=lfs diff=lfs merge=lfs -text
+val_batch1_pred.jpg filter=lfs diff=lfs merge=lfs -text
+val_batch2_labels.jpg filter=lfs diff=lfs merge=lfs -text
+val_batch2_pred.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch7110.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch7111.jpg filter=lfs diff=lfs merge=lfs -text
+train_batch7112.jpg filter=lfs diff=lfs merge=lfs -text

args.yaml

@@ -0,0 +1,106 @@
+task: detect
+mode: train
+model: yolov8n.pt
+data: configs/ppe_dataset.yaml
+epochs: 100
+time: null
+patience: 20
+batch: 128
+imgsz: 640
+save: true
+save_period: -1
+cache: false
+device: '0'
+workers: 8
+project: models
+name: ppe_detection
+exist_ok: true
+pretrained: true
+optimizer: AdamW
+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
+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
+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: /workspace/SafetyVisionAI/models/ppe_detection

results.csv

@@ -0,0 +1,101 @@
+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,51.4123,1.50065,1.8234,1.37511,0.37293,0.09591,0.04147,0.0177,2.43915,5.47621,2.95854,0.0703797,0.00329114,0.00329114
+2,84.1956,1.46978,1.35194,1.35969,0.25994,0.20463,0.16402,0.07136,2.28536,2.67931,2.46817,0.0403142,0.00655889,0.00655889
+3,117.158,1.46793,1.31559,1.37515,0.43985,0.37087,0.31265,0.13297,2.36465,2.83757,2.41708,0.0101826,0.00976064,0.00976064
+4,156.895,1.41754,1.22273,1.34258,0.40146,0.34531,0.28762,0.14019,2.08963,4.74887,2.20895,0.009703,0.009703,0.009703
+5,191.664,1.36586,1.12516,1.30837,0.61875,0.25967,0.24496,0.11895,2.03598,2.94078,2.10611,0.009604,0.009604,0.009604
+6,225.628,1.32394,1.0711,1.28322,0.75782,0.37937,0.43785,0.241,1.62271,1.68313,1.55079,0.009505,0.009505,0.009505
+7,260.119,1.31868,1.03334,1.27489,0.7585,0.69827,0.76036,0.44838,1.40086,1.17108,1.38321,0.009406,0.009406,0.009406
+8,295.185,1.28346,0.99455,1.26095,0.74503,0.44493,0.54675,0.28909,1.65879,1.64064,1.61566,0.009307,0.009307,0.009307
+9,328.01,1.24333,0.93823,1.22642,0.82857,0.71117,0.81667,0.50612,1.27159,0.94526,1.26202,0.009208,0.009208,0.009208
+10,360.965,1.22362,0.91305,1.21412,0.79432,0.70059,0.79651,0.47206,1.35862,0.9783,1.3153,0.009109,0.009109,0.009109
+11,393.962,1.22276,0.90452,1.21442,0.77695,0.54842,0.66212,0.39494,1.45233,1.11845,1.43626,0.00901,0.00901,0.00901
+12,427.579,1.20647,0.87411,1.20268,0.83039,0.75708,0.82791,0.51277,1.25437,0.9173,1.25986,0.008911,0.008911,0.008911
+13,461.033,1.20114,0.87169,1.20034,0.80964,0.72174,0.80295,0.48711,1.33708,0.99435,1.32541,0.008812,0.008812,0.008812
+14,494.021,1.18221,0.84588,1.18942,0.82554,0.73802,0.81649,0.5234,1.21503,0.89161,1.23147,0.008713,0.008713,0.008713
+15,527.389,1.18521,0.84175,1.19089,0.82413,0.76631,0.82835,0.52639,1.25351,0.8458,1.24172,0.008614,0.008614,0.008614
+16,561.69,1.18022,0.83273,1.18836,0.84747,0.74932,0.84514,0.53975,1.23379,0.85808,1.22781,0.008515,0.008515,0.008515
+17,595.269,1.16232,0.81387,1.17908,0.85922,0.79823,0.8731,0.57508,1.15522,0.73717,1.16875,0.008416,0.008416,0.008416
+18,627.558,1.14521,0.796,1.16524,0.85022,0.78932,0.86299,0.5507,1.22577,0.7735,1.22144,0.008317,0.008317,0.008317
+19,660.005,1.14288,0.78269,1.16173,0.84923,0.79399,0.86869,0.56916,1.17494,0.73094,1.19261,0.008218,0.008218,0.008218
+20,692.378,1.13581,0.78047,1.15587,0.80853,0.57993,0.73937,0.42392,1.404,1.00321,1.41002,0.008119,0.008119,0.008119
+21,724.569,1.1292,0.77581,1.1599,0.85148,0.7674,0.85387,0.53783,1.24207,0.83228,1.23964,0.00802,0.00802,0.00802
+22,756.622,1.12095,0.76122,1.14985,0.85788,0.79221,0.86212,0.57129,1.14413,0.75348,1.17286,0.007921,0.007921,0.007921
+23,788.618,1.11085,0.749,1.1395,0.88335,0.81339,0.8912,0.60703,1.10543,0.67025,1.14189,0.007822,0.007822,0.007822
+24,820.862,1.11583,0.74928,1.14697,0.85815,0.78362,0.86737,0.57935,1.13867,0.73767,1.17339,0.007723,0.007723,0.007723
+25,852.774,1.11108,0.74442,1.14799,0.7819,0.61504,0.69786,0.42754,1.39154,1.03464,1.38793,0.007624,0.007624,0.007624
+26,885.063,1.09913,0.73488,1.13873,0.82516,0.72297,0.7962,0.49881,1.25708,0.89321,1.28852,0.007525,0.007525,0.007525
+27,917.888,1.08664,0.72678,1.1278,0.80441,0.76254,0.84116,0.5592,1.16084,0.77638,1.19387,0.007426,0.007426,0.007426
+28,950.477,1.09462,0.72379,1.13289,0.88215,0.83744,0.90637,0.60866,1.13647,0.65935,1.1567,0.007327,0.007327,0.007327
+29,982.846,1.08668,0.72372,1.12821,0.88561,0.83369,0.90826,0.62445,1.07424,0.63411,1.1136,0.007228,0.007228,0.007228
+30,1015.53,1.08425,0.70726,1.12619,0.88714,0.82971,0.90702,0.62195,1.07262,0.64834,1.12683,0.007129,0.007129,0.007129
+31,1048.09,1.07391,0.70453,1.12612,0.82842,0.77608,0.84443,0.54359,1.21963,0.78714,1.2597,0.00703,0.00703,0.00703
+32,1081.22,1.06582,0.68921,1.11709,0.90151,0.84594,0.91983,0.63735,1.04893,0.59596,1.10132,0.006931,0.006931,0.006931
+33,1114.25,1.05936,0.69157,1.11604,0.89329,0.83645,0.91355,0.63572,1.06716,0.62406,1.11004,0.006832,0.006832,0.006832
+34,1147.46,1.06039,0.6922,1.11451,0.88782,0.82209,0.89755,0.62233,1.07067,0.6384,1.11634,0.006733,0.006733,0.006733
+35,1179.46,1.05347,0.68141,1.10957,0.8901,0.85784,0.9179,0.64075,1.04885,0.6013,1.104,0.006634,0.006634,0.006634
+36,1211.56,1.05659,0.68006,1.11203,0.90157,0.84512,0.91475,0.63379,1.06495,0.60758,1.1073,0.006535,0.006535,0.006535
+37,1243.73,1.04628,0.66755,1.10606,0.81188,0.74713,0.82351,0.5406,1.19184,0.77537,1.23455,0.006436,0.006436,0.006436
+38,1275.91,1.04476,0.66943,1.10876,0.89697,0.84583,0.91791,0.6412,1.0392,0.61888,1.09669,0.006337,0.006337,0.006337
+39,1308.24,1.039,0.66963,1.10684,0.8089,0.58428,0.68069,0.42641,1.36045,0.97447,1.39825,0.006238,0.006238,0.006238
+40,1339.98,1.03723,0.6596,1.09871,0.81639,0.72094,0.82153,0.54026,1.22374,0.84343,1.2718,0.006139,0.006139,0.006139
+41,1372.15,1.03464,0.65641,1.10036,0.89202,0.86163,0.91909,0.64665,1.02944,0.59257,1.09586,0.00604,0.00604,0.00604
+42,1405.05,1.03149,0.65654,1.09654,0.80347,0.68474,0.79492,0.5167,1.23706,0.86262,1.27294,0.005941,0.005941,0.005941
+43,1436.92,1.02789,0.64833,1.09497,0.90208,0.84882,0.92103,0.64891,1.03776,0.58573,1.09949,0.005842,0.005842,0.005842
+44,1469.46,1.0234,0.64746,1.09341,0.82238,0.75096,0.82758,0.52687,1.27553,0.79161,1.31339,0.005743,0.005743,0.005743
+45,1502.04,1.02193,0.64122,1.08869,0.89146,0.85294,0.91876,0.6426,1.04213,0.57602,1.08632,0.005644,0.005644,0.005644
+46,1533.95,1.01302,0.63484,1.08758,0.90437,0.86028,0.92304,0.65219,1.01906,0.56817,1.09258,0.005545,0.005545,0.005545
+47,1566.09,1.01707,0.63359,1.09039,0.88393,0.86336,0.91649,0.64787,1.02484,0.57998,1.0916,0.005446,0.005446,0.005446
+48,1598.21,1.01897,0.63319,1.08745,0.90962,0.85466,0.9253,0.65603,1.00723,0.5609,1.07854,0.005347,0.005347,0.005347
+49,1630.38,1.00241,0.62226,1.08692,0.87617,0.83168,0.895,0.62364,1.04814,0.6111,1.11974,0.005248,0.005248,0.005248
+50,1662.44,0.9968,0.61978,1.0795,0.88096,0.83985,0.90032,0.62167,1.06316,0.63198,1.12867,0.005149,0.005149,0.005149
+51,1694.5,1.00311,0.61608,1.08217,0.90895,0.85689,0.9282,0.65794,1.01444,0.55126,1.07989,0.00505,0.00505,0.00505
+52,1726.54,0.99102,0.6094,1.07559,0.87808,0.79968,0.88108,0.60738,1.10527,0.67443,1.1541,0.004951,0.004951,0.004951
+53,1758.47,0.99572,0.61405,1.07493,0.91448,0.86276,0.9328,0.66667,0.98818,0.53537,1.06131,0.004852,0.004852,0.004852
+54,1790.65,0.995,0.61596,1.07949,0.90457,0.86565,0.92978,0.6618,1.00197,0.54342,1.07233,0.004753,0.004753,0.004753
+55,1822.54,0.99045,0.60682,1.07491,0.90906,0.86951,0.93171,0.66938,0.98413,0.52893,1.05559,0.004654,0.004654,0.004654
+56,1855.08,0.98454,0.59832,1.07036,0.9153,0.86523,0.93249,0.67185,0.98225,0.52555,1.05909,0.004555,0.004555,0.004555
+57,1887.34,0.98273,0.59893,1.06876,0.88022,0.82441,0.89612,0.626,1.06503,0.62851,1.13071,0.004456,0.004456,0.004456
+58,1919.35,0.97773,0.59244,1.06706,0.84624,0.78648,0.85901,0.58173,1.1557,0.70329,1.22199,0.004357,0.004357,0.004357
+59,1951.26,0.97706,0.59305,1.06794,0.89046,0.84337,0.91164,0.64662,1.01344,0.5753,1.09661,0.004258,0.004258,0.004258
+60,1983.31,0.98335,0.59242,1.06656,0.89751,0.85969,0.92273,0.66098,0.99509,0.5507,1.07468,0.004159,0.004159,0.004159
+61,2015.21,0.96689,0.58538,1.06064,0.91101,0.8755,0.93663,0.67382,0.97365,0.51589,1.04864,0.00406,0.00406,0.00406
+62,2047.67,0.97375,0.584,1.06429,0.90743,0.87788,0.93489,0.6751,0.9661,0.51466,1.05108,0.003961,0.003961,0.003961
+63,2079.71,0.96875,0.57964,1.06095,0.86214,0.80037,0.87551,0.60334,1.10818,0.64758,1.17188,0.003862,0.003862,0.003862
+64,2112.07,0.96219,0.57936,1.06097,0.90197,0.85587,0.92185,0.66492,0.9896,0.54504,1.07446,0.003763,0.003763,0.003763
+65,2143.99,0.95529,0.57691,1.06122,0.80572,0.72185,0.81644,0.54089,1.20487,0.76997,1.27183,0.003664,0.003664,0.003664
+66,2176,0.95582,0.57611,1.05688,0.9059,0.88077,0.93462,0.67661,0.96635,0.5092,1.04876,0.003565,0.003565,0.003565
+67,2208.17,0.95036,0.57095,1.05399,0.91624,0.87785,0.93734,0.68049,0.96091,0.50435,1.04526,0.003466,0.003466,0.003466
+68,2240.11,0.95182,0.57139,1.05341,0.89732,0.87657,0.92757,0.6668,0.98182,0.5249,1.06494,0.003367,0.003367,0.003367
+69,2272.01,0.95039,0.56666,1.05483,0.89728,0.86434,0.92219,0.66274,0.99347,0.54079,1.07542,0.003268,0.003268,0.003268
+70,2303.57,0.94957,0.56265,1.05446,0.80065,0.68895,0.7813,0.51348,1.25645,0.83648,1.33046,0.003169,0.003169,0.003169
+71,2335.2,0.94737,0.5586,1.05254,0.89909,0.8805,0.93324,0.67831,0.96663,0.51548,1.05145,0.00307,0.00307,0.00307
+72,2367.1,0.93982,0.55988,1.04597,0.91319,0.88129,0.93874,0.68284,0.95778,0.49888,1.04712,0.002971,0.002971,0.002971
+73,2399,0.93673,0.55444,1.04799,0.89325,0.85627,0.91799,0.65918,1.0017,0.54082,1.08151,0.002872,0.002872,0.002872
+74,2431.12,0.93363,0.54845,1.04491,0.89769,0.85714,0.91552,0.65811,1.00025,0.54298,1.08961,0.002773,0.002773,0.002773
+75,2462.96,0.93278,0.54608,1.04521,0.91501,0.87672,0.93596,0.68323,0.95503,0.49869,1.04106,0.002674,0.002674,0.002674
+76,2494.77,0.92856,0.54541,1.03924,0.91145,0.87591,0.93344,0.68045,0.96126,0.50223,1.04768,0.002575,0.002575,0.002575
+77,2526.26,0.92669,0.54621,1.04084,0.87499,0.82203,0.89356,0.62804,1.0573,0.59537,1.13399,0.002476,0.002476,0.002476
+78,2557.81,0.9246,0.54272,1.04025,0.91932,0.87976,0.9394,0.68761,0.94752,0.48554,1.0414,0.002377,0.002377,0.002377
+79,2589.46,0.92002,0.53982,1.03802,0.91934,0.87742,0.93868,0.68599,0.94916,0.49069,1.04406,0.002278,0.002278,0.002278
+80,2620.87,0.91669,0.53521,1.03751,0.91595,0.87717,0.93804,0.68395,0.95605,0.49773,1.04637,0.002179,0.002179,0.002179
+81,2652.77,0.91689,0.53542,1.03545,0.90459,0.8714,0.92945,0.6757,0.966,0.51241,1.05437,0.00208,0.00208,0.00208
+82,2684.29,0.90648,0.52599,1.03491,0.88576,0.85667,0.91328,0.6559,1.00095,0.55201,1.09114,0.001981,0.001981,0.001981
+83,2716.77,0.91159,0.52621,1.03115,0.91848,0.87586,0.93846,0.68624,0.95303,0.48943,1.04326,0.001882,0.001882,0.001882
+84,2748.73,0.91059,0.52874,1.0317,0.92153,0.87954,0.93951,0.68863,0.94586,0.48073,1.03489,0.001783,0.001783,0.001783
+85,2780.78,0.90694,0.52404,1.02979,0.92248,0.88244,0.94197,0.69023,0.942,0.48344,1.03617,0.001684,0.001684,0.001684
+86,2813.49,0.90004,0.52174,1.02878,0.92112,0.87456,0.93906,0.68899,0.94619,0.48613,1.04055,0.001585,0.001585,0.001585
+87,2845.47,0.90256,0.5183,1.02863,0.91185,0.87414,0.93305,0.68116,0.9578,0.49905,1.05238,0.001486,0.001486,0.001486
+88,2877.26,0.90254,0.5187,1.03113,0.91481,0.87369,0.93613,0.68446,0.95474,0.49237,1.04795,0.001387,0.001387,0.001387
+89,2909.37,0.89562,0.51332,1.02731,0.91766,0.87449,0.93767,0.68717,0.95034,0.49022,1.04504,0.001288,0.001288,0.001288
+90,2941.52,0.89204,0.51187,1.02581,0.92272,0.88103,0.94124,0.69236,0.9404,0.47695,1.03327,0.001189,0.001189,0.001189
+91,2976.39,0.85934,0.44024,1.00545,0.9217,0.88344,0.94143,0.69245,0.93796,0.47505,1.03226,0.00109,0.00109,0.00109
+92,3008.78,0.8502,0.42653,0.99989,0.91975,0.88375,0.9396,0.6902,0.9423,0.48192,1.03778,0.000991,0.000991,0.000991
+93,3040.48,0.84569,0.42471,0.99954,0.9118,0.8823,0.93655,0.68797,0.94922,0.48947,1.04613,0.000892,0.000892,0.000892
+94,3072.4,0.84105,0.42021,0.99858,0.92452,0.88263,0.9397,0.69191,0.93857,0.47738,1.03746,0.000793,0.000793,0.000793
+95,3104.01,0.83442,0.41802,0.99073,0.90853,0.87677,0.93392,0.6857,0.95205,0.49706,1.04984,0.000694,0.000694,0.000694
+96,3135.55,0.82803,0.41205,0.99035,0.90599,0.87673,0.93449,0.68546,0.95048,0.49505,1.04856,0.000595,0.000595,0.000595
+97,3167.36,0.8272,0.41369,0.98788,0.91793,0.87232,0.93728,0.68934,0.94419,0.48682,1.04161,0.000496,0.000496,0.000496
+98,3199.31,0.82701,0.41139,0.98693,0.91477,0.87618,0.93624,0.68839,0.94696,0.49006,1.04413,0.000397,0.000397,0.000397
+99,3231.04,0.81711,0.40693,0.98658,0.91909,0.88005,0.93928,0.69331,0.93824,0.48007,1.03709,0.000298,0.000298,0.000298
+100,3262.47,0.82135,0.40759,0.98722,0.92227,0.87219,0.93677,0.68952,0.94606,0.48846,1.04497,0.000199,0.000199,0.000199

weights/best.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8aa1fdcf6f842a0d7bee8b782d120df337284e6f5d1fb4fe37cdcd7aaf42d01e
+size 6257827

weights/last.pt

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:c3f8f54ef20fe9091e8ace8c9176dbc315368e98d16efe3dacf77c1130558821
+size 6257827