Adding v1

.gitattributes

@@ -33,3 +33,4 @@ 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
+Box*.png filter=lfs diff=lfs merge=lfs -text

README.md

@@ -1,3 +0,0 @@
----
-license: mit
----

v1/billiards-yolov8n.onnx

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

v1/billiards-yolov8n.pt

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

v1/config.json

@@ -0,0 +1,12 @@
+{
+  "model_name": "billiards-yolov8n",
+  "format": "onnx",
+  "input_size": 640,
+  "framework": "YOLOv8-nano",
+  "classes": ["cue", "ball"],
+  "num_classes": 3,
+  "pretrained": false,
+  "training_dataset": "https://huggingface.co/datasets/jchadwick/billiards",
+  "version": "v1"
+}
+

v1/metadata/args.yaml

@@ -0,0 +1,106 @@
+task: detect
+mode: train
+model: yolov8n.pt
+data: /Users/jchadwick/code/billiards-trainer/dataset/data.yaml
+epochs: 100
+time: null
+patience: 20
+batch: 8
+imgsz: 640
+save: true
+save_period: -1
+cache: false
+device: mps
+workers: 8
+project: /Users/jchadwick/code/billiards-trainer/models/training_runs
+name: yolov8n_pool_v117
+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
+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: /Users/jchadwick/code/billiards-trainer/models/training_runs/yolov8n_pool_v117

v1/metadata/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,18.238,1.52672,3.18222,1.03726,0.02299,0.96078,0.25135,0.16479,1.09072,3.58376,0.88294,0.00026672,0.00026672,0.00026672
+2,33.5118,1.25462,1.45531,0.94399,0.02024,0.96324,0.27773,0.14301,1.06316,3.24082,0.90532,0.000544664,0.000544664,0.000544664
+3,49.1327,1.20527,1.26931,0.93901,0.02039,0.96814,0.57149,0.39352,0.98674,2.74204,0.91086,0.000816997,0.000816997,0.000816997
+4,65.1896,1.1038,1.06351,0.90236,0.03447,0.83565,0.70151,0.46107,1.01911,2.57279,0.92145,0.00108372,0.00108372,0.00108372
+5,82.2928,1.10576,1.07686,0.93674,0.75458,0.60302,0.78486,0.53025,0.97907,1.87712,0.90921,0.00134483,0.00134483,0.00134483
+6,98.6127,1.22434,1.0802,0.94,0.52778,0.80081,0.74193,0.5662,0.88235,1.60612,0.89873,0.00158448,0.00158448,0.00158448
+7,114.481,1.12597,1.06522,0.92069,0.75596,0.77104,0.75846,0.57591,0.84584,1.06985,0.88573,0.00156798,0.00156798,0.00156798
+8,129.798,1.11474,1.04563,0.93382,0.58379,0.89674,0.80744,0.5886,0.95985,1.21068,0.87198,0.00155148,0.00155148,0.00155148
+9,146.259,1.06594,0.97114,0.90597,0.63178,0.84547,0.79511,0.57645,0.86092,0.97709,0.86814,0.00153497,0.00153497,0.00153497
+10,164.479,1.05814,0.98096,0.90023,0.95632,0.67402,0.77758,0.58502,0.79572,0.82688,0.87016,0.00151847,0.00151847,0.00151847
+11,184.218,0.97211,0.93443,0.90815,0.69761,0.78422,0.76398,0.57254,0.75168,0.95063,0.87817,0.00150197,0.00150197,0.00150197
+12,204.156,1.04606,0.93431,0.91427,0.73512,0.79064,0.77931,0.58838,0.76714,1.06662,0.87299,0.00148546,0.00148546,0.00148546
+13,220.968,0.92527,0.84609,0.89875,0.68003,0.73284,0.74858,0.60606,0.71611,0.93564,0.87973,0.00146896,0.00146896,0.00146896
+14,241.26,0.95564,0.82119,0.89306,0.77764,0.76076,0.73135,0.57795,0.72213,0.80295,0.8873,0.00145246,0.00145246,0.00145246
+15,261.308,0.93598,0.80734,0.88646,0.75443,0.79167,0.76923,0.59592,0.72723,0.76991,0.87828,0.00143595,0.00143595,0.00143595
+16,280.204,0.93109,0.7695,0.89325,0.75659,0.87607,0.81415,0.63129,0.70881,0.70537,0.87112,0.00141945,0.00141945,0.00141945
+17,300.04,0.90133,0.79766,0.88848,0.89113,0.81709,0.79105,0.63274,0.72448,0.66702,0.86629,0.00140295,0.00140295,0.00140295
+18,319.744,0.88773,0.77814,0.88267,0.90882,0.82108,0.8249,0.64647,0.68577,0.66584,0.8787,0.00138644,0.00138644,0.00138644
+19,338.814,0.89215,0.73913,0.87067,0.78211,0.90931,0.8737,0.63301,0.77089,0.64008,0.8708,0.00136994,0.00136994,0.00136994
+20,357.744,0.89598,0.70223,0.87344,0.86077,0.9624,0.90001,0.62753,0.66619,0.62326,0.87121,0.00135344,0.00135344,0.00135344
+21,375.52,0.89272,0.69772,0.88517,0.79019,0.92429,0.85317,0.65094,0.67433,0.61887,0.86532,0.00133693,0.00133693,0.00133693
+22,393.751,0.82414,0.65823,0.86569,0.81354,0.88395,0.85483,0.64147,0.6808,0.57171,0.86455,0.00132043,0.00132043,0.00132043
+23,415.589,0.87107,0.69722,0.87833,0.93005,0.76225,0.83358,0.62115,0.65569,0.55128,0.87033,0.00130393,0.00130393,0.00130393
+24,438.385,0.77657,0.62796,0.85793,0.88913,0.76225,0.82238,0.58361,0.65572,0.5614,0.87148,0.00128742,0.00128742,0.00128742
+25,458.737,0.81272,0.66724,0.86775,0.88751,0.90931,0.94357,0.66939,0.72028,0.57911,0.87702,0.00127092,0.00127092,0.00127092
+26,480.022,0.87186,0.70849,0.88085,0.9563,0.85029,0.90687,0.65909,0.70595,0.56959,0.87525,0.00125442,0.00125442,0.00125442
+27,502.711,0.82476,0.67554,0.88163,0.92916,0.79167,0.88254,0.69424,0.70503,0.55562,0.87066,0.00123791,0.00123791,0.00123791
+28,524.85,0.84192,0.64305,0.86554,0.82868,0.90931,0.87951,0.66433,0.63686,0.54425,0.86537,0.00122141,0.00122141,0.00122141
+29,547.022,0.78151,0.63979,0.86078,0.91921,0.81752,0.89341,0.67013,0.63389,0.53389,0.86535,0.00120491,0.00120491,0.00120491
+30,570.154,0.81679,0.59982,0.85396,0.86229,0.89645,0.90503,0.68256,0.64753,0.53454,0.86195,0.0011884,0.0011884,0.0011884
+31,590.568,0.86994,0.6421,0.86216,0.8656,0.90343,0.89677,0.66971,0.66308,0.51446,0.86407,0.0011719,0.0011719,0.0011719
+32,612.541,0.79704,0.62084,0.86176,0.90925,0.90931,0.92349,0.70855,0.62341,0.51586,0.8562,0.0011554,0.0011554,0.0011554
+33,635.046,0.78398,0.58908,0.84785,0.96958,0.79167,0.83938,0.61975,0.70127,0.55245,0.8563,0.00113889,0.00113889,0.00113889
+34,654.829,0.74044,0.57323,0.84829,0.94588,0.90931,0.93764,0.72751,0.60298,0.51915,0.85483,0.00112239,0.00112239,0.00112239
+35,677.684,0.77024,0.58023,0.85057,0.90688,0.85049,0.86095,0.6802,0.60264,0.50214,0.85724,0.00110589,0.00110589,0.00110589
+36,701.316,0.75665,0.55499,0.84192,0.90638,0.84319,0.84053,0.64736,0.61015,0.4989,0.85552,0.00108938,0.00108938,0.00108938
+37,725.661,0.77311,0.59413,0.85957,0.94851,0.82108,0.87277,0.64883,0.58787,0.4781,0.85481,0.00107288,0.00107288,0.00107288
+38,750.689,0.72886,0.55524,0.85633,0.85465,0.73529,0.74366,0.55014,0.62716,0.47452,0.8511,0.00105638,0.00105638,0.00105638
+39,777.799,0.74687,0.55064,0.84684,0.90136,0.67892,0.73199,0.52171,0.63642,0.48732,0.85247,0.00103987,0.00103987,0.00103987
+40,803.827,0.74495,0.56574,0.84057,0.86404,0.8799,0.85644,0.61833,0.65004,0.50272,0.85518,0.00102337,0.00102337,0.00102337
+41,828.216,0.82433,0.58533,0.85494,0.89417,0.90931,0.91468,0.67113,0.68866,0.51583,0.86285,0.00100687,0.00100687,0.00100687
+42,853.964,0.7634,0.55192,0.84677,0.89506,0.90931,0.90254,0.71692,0.61813,0.49883,0.8601,0.000990365,0.000990365,0.000990365
+43,879.596,0.76873,0.57032,0.85765,0.93128,0.90636,0.92719,0.73208,0.61487,0.49313,0.86369,0.000973861,0.000973861,0.000973861
+44,904.547,0.78946,0.56197,0.84511,0.91299,0.95524,0.93558,0.73174,0.60181,0.46589,0.85646,0.000957358,0.000957358,0.000957358
+45,929.987,0.78674,0.54545,0.84282,0.90356,0.93873,0.92679,0.69477,0.70179,0.48879,0.86033,0.000940855,0.000940855,0.000940855
+46,955.111,0.79916,0.55454,0.85055,0.89731,0.93873,0.90884,0.6972,0.58471,0.45531,0.86067,0.000924352,0.000924352,0.000924352
+47,979.234,0.77352,0.55377,0.84256,0.91574,0.962,0.94354,0.69756,0.61197,0.4531,0.85689,0.000907848,0.000907848,0.000907848
+48,1004.38,0.72131,0.53139,0.84577,0.96529,0.9023,0.9519,0.73396,0.55129,0.44092,0.85404,0.000891345,0.000891345,0.000891345
+49,1027.2,0.72147,0.51352,0.83998,0.95232,0.90931,0.92761,0.7414,0.62692,0.44962,0.84998,0.000874842,0.000874842,0.000874842
+50,1049.65,0.79546,0.52983,0.8444,0.91207,0.90931,0.91298,0.74956,0.60198,0.4453,0.85512,0.000858338,0.000858338,0.000858338
+51,1071.26,0.7639,0.52942,0.84829,0.92251,0.90931,0.92609,0.70209,0.7181,0.49,0.85879,0.000841835,0.000841835,0.000841835
+52,1093.52,0.8021,0.54985,0.84297,0.96435,0.90064,0.93762,0.73351,0.60919,0.46358,0.85914,0.000825332,0.000825332,0.000825332
+53,1114.04,0.79592,0.52441,0.83236,0.9525,0.90931,0.93975,0.73983,0.62409,0.4417,0.856,0.000808828,0.000808828,0.000808828
+54,1135.78,0.73423,0.51819,0.85321,0.96383,0.93551,0.95145,0.7545,0.54209,0.43842,0.85549,0.000792325,0.000792325,0.000792325
+55,1158.18,0.69602,0.50454,0.84176,0.96364,0.93123,0.947,0.74668,0.58578,0.43463,0.85456,0.000775822,0.000775822,0.000775822
+56,1181.01,0.71624,0.50076,0.85107,0.9543,0.90931,0.93368,0.75145,0.57147,0.42239,0.85523,0.000759318,0.000759318,0.000759318
+57,1202.74,0.72321,0.51476,0.84041,0.96032,0.87654,0.92934,0.7518,0.56582,0.42086,0.85446,0.000742815,0.000742815,0.000742815
+58,1225.53,0.69306,0.50507,0.83765,0.85453,0.93873,0.92105,0.75434,0.53531,0.42237,0.84963,0.000726312,0.000726312,0.000726312
+59,1247.89,0.65692,0.47877,0.83624,0.93101,0.93873,0.92422,0.74711,0.55508,0.41896,0.85051,0.000709809,0.000709809,0.000709809
+60,1277.71,0.6888,0.50441,0.84438,0.93186,0.93873,0.92369,0.75042,0.55223,0.41227,0.85229,0.000693305,0.000693305,0.000693305
+61,1307.02,0.69936,0.48372,0.8286,0.86594,0.93873,0.91894,0.73827,0.5468,0.42376,0.8535,0.000676802,0.000676802,0.000676802
+62,1332.34,0.71993,0.48862,0.83862,0.86749,0.91431,0.92508,0.74244,0.53887,0.41144,0.85374,0.000660299,0.000660299,0.000660299
+63,1360.29,0.69542,0.48755,0.84132,0.88678,0.92019,0.92079,0.74334,0.53406,0.40957,0.8512,0.000643795,0.000643795,0.000643795
+64,1384.89,0.72596,0.50928,0.85091,0.90513,0.93873,0.92536,0.73929,0.54095,0.41073,0.85389,0.000627292,0.000627292,0.000627292
+65,1410.79,0.6758,0.47205,0.83246,0.95701,0.90931,0.93576,0.75886,0.5409,0.413,0.85878,0.000610789,0.000610789,0.000610789
+66,1436.91,0.67785,0.47027,0.83183,0.95312,0.90931,0.93821,0.76995,0.5234,0.41195,0.85008,0.000594286,0.000594286,0.000594286
+67,1464.23,0.67142,0.4624,0.83285,0.99635,0.85049,0.88907,0.70118,0.52055,0.4268,0.84761,0.000577782,0.000577782,0.000577782
+68,1490.44,0.64606,0.47153,0.82922,0.91473,0.93873,0.95198,0.76196,0.5148,0.40655,0.8526,0.000561279,0.000561279,0.000561279
+69,1514.36,0.69085,0.46783,0.83819,0.91353,0.93873,0.95086,0.75033,0.52508,0.40402,0.85072,0.000544776,0.000544776,0.000544776
+70,1534.9,0.66631,0.47029,0.83226,0.93234,0.93873,0.95907,0.76898,0.49445,0.3919,0.847,0.000528272,0.000528272,0.000528272
+71,1557.7,0.648,0.46061,0.82599,0.94747,0.93873,0.95436,0.7744,0.49708,0.39311,0.84427,0.000511769,0.000511769,0.000511769
+72,1579.95,0.62871,0.44082,0.83402,0.96616,0.92444,0.94855,0.77087,0.4892,0.38359,0.84473,0.000495266,0.000495266,0.000495266
+73,1602.7,0.66553,0.45742,0.82204,0.90916,0.93873,0.93985,0.77798,0.51522,0.39725,0.84877,0.000478762,0.000478762,0.000478762
+74,1625.75,0.67346,0.46176,0.83944,0.90894,0.93873,0.93131,0.75999,0.5149,0.39544,0.84807,0.000462259,0.000462259,0.000462259
+75,1646.66,0.67285,0.48238,0.83334,0.91015,0.93873,0.92908,0.7492,0.51567,0.40905,0.84528,0.000445756,0.000445756,0.000445756
+76,1666.82,0.61813,0.43949,0.81982,0.91137,0.93616,0.9279,0.75984,0.50145,0.38704,0.843,0.000429253,0.000429253,0.000429253
+77,1689.06,0.61836,0.4318,0.81128,0.9091,0.90931,0.93299,0.75615,0.49503,0.39154,0.8421,0.000412749,0.000412749,0.000412749
+78,1712.39,0.62045,0.43214,0.82259,0.90797,0.90931,0.92183,0.76862,0.48675,0.38088,0.84309,0.000396246,0.000396246,0.000396246
+79,1737.47,0.62702,0.44212,0.82331,0.92692,0.90931,0.91273,0.75723,0.49994,0.37864,0.8439,0.000379743,0.000379743,0.000379743
+80,1758.4,0.6579,0.45408,0.83372,0.91342,0.93102,0.92214,0.75797,0.5075,0.37493,0.84528,0.000363239,0.000363239,0.000363239
+81,1779.92,0.59777,0.4276,0.82045,0.90888,0.93873,0.92994,0.75313,0.54735,0.38362,0.84738,0.000346736,0.000346736,0.000346736
+82,1802.85,0.64959,0.45196,0.82536,0.93224,0.93873,0.93595,0.76188,0.52579,0.37472,0.8449,0.000330233,0.000330233,0.000330233
+83,1823.8,0.61858,0.43156,0.81864,0.92571,0.93873,0.94056,0.76837,0.50838,0.37434,0.84476,0.000313729,0.000313729,0.000313729
+84,1841.66,0.62824,0.4174,0.82681,0.92304,0.93873,0.94105,0.77938,0.49702,0.36906,0.84384,0.000297226,0.000297226,0.000297226
+85,1859.79,0.63772,0.4264,0.82382,0.93659,0.93873,0.94997,0.78286,0.47913,0.36062,0.84271,0.000280723,0.000280723,0.000280723
+86,1879.84,0.64064,0.45281,0.83456,0.94509,0.93873,0.95017,0.78903,0.50646,0.37153,0.84209,0.00026422,0.00026422,0.00026422
+87,1901.34,0.60332,0.4256,0.83379,0.94756,0.93873,0.94899,0.78599,0.49074,0.36952,0.84239,0.000247716,0.000247716,0.000247716
+88,1920.36,0.61487,0.43986,0.8298,0.94573,0.93873,0.95047,0.79013,0.48918,0.37576,0.84193,0.000231213,0.000231213,0.000231213
+89,1939.11,0.62031,0.42652,0.827,0.95185,0.93873,0.95033,0.80152,0.47804,0.3645,0.8427,0.00021471,0.00021471,0.00021471
+90,1956.73,0.63882,0.43281,0.82888,0.94015,0.93873,0.94648,0.80525,0.4653,0.36913,0.84264,0.000198206,0.000198206,0.000198206
+91,1979.55,0.56734,0.40872,0.81592,0.93569,0.93873,0.94563,0.78024,0.49828,0.36295,0.84062,0.000181703,0.000181703,0.000181703
+92,2003.37,0.60466,0.42324,0.81891,0.93509,0.93873,0.94586,0.7829,0.4935,0.36106,0.83999,0.0001652,0.0001652,0.0001652
+93,2022.63,0.54947,0.38769,0.80275,0.9348,0.93873,0.94495,0.78247,0.47178,0.35404,0.84419,0.000148696,0.000148696,0.000148696
+94,2043.21,0.5651,0.40266,0.80471,0.93427,0.93873,0.94337,0.77597,0.48682,0.3527,0.84418,0.000132193,0.000132193,0.000132193
+95,2063.2,0.53918,0.37775,0.80966,0.93343,0.93873,0.93827,0.77777,0.48091,0.35337,0.84283,0.00011569,0.00011569,0.00011569
+96,2086.45,0.55524,0.3886,0.80402,0.93347,0.93873,0.94116,0.78781,0.47578,0.35139,0.84122,9.91865e-05,9.91865e-05,9.91865e-05
+97,2107.63,0.58061,0.41863,0.81347,0.93453,0.93873,0.94719,0.79926,0.4712,0.34594,0.83996,8.26832e-05,8.26832e-05,8.26832e-05
+98,2129.31,0.5539,0.39128,0.79664,0.92108,0.93873,0.9453,0.79902,0.47379,0.35036,0.8403,6.61799e-05,6.61799e-05,6.61799e-05
+99,2150.14,0.57699,0.40439,0.8093,0.93584,0.91647,0.94816,0.80177,0.46974,0.35161,0.8399,4.96766e-05,4.96766e-05,4.96766e-05
+100,2169.63,0.55874,0.40091,0.81285,0.93765,0.93873,0.94786,0.79886,0.46377,0.34901,0.83946,3.31733e-05,3.31733e-05,3.31733e-05