Model save

config.json

@@ -38,263 +38,21 @@
   "giou_cost": 2,
   "giou_loss_coefficient": 2,
   "id2label": {
-    "0": "grey and white light aircraft",
-    "1": "grey fighter jet",
-    "2": "grey helicopter",
-    "3": "orange and black fighter jet",
-    "4": "white commercial aircraft",
-    "5": "white and yellow commercial aircraft",
-    "6": "white light aircraft",
-    "7": "blue and green fighter plane",
-    "8": "green fighter plane",
-    "9": "white, blue, and red commercial aircraft",
-    "10": "white and blue fighter plane",
-    "11": "yellow and red light aircraft",
-    "12": "grey drone",
-    "13": "black camouflage fighter jet",
-    "14": "blue and red light aircraft",
-    "15": "black and white commercial aircraft",
-    "16": "blue commercial aircraft",
-    "17": "black cargo aircraft",
-    "18": "yellow, black, and red helicopter",
-    "19": "white, red, and green fighter plane",
-    "20": "white and blue fighter jet",
-    "21": "blue and grey fighter jet",
-    "22": "green and grey helicopter",
-    "23": "blue and red commercial aircraft",
-    "24": "yellow fighter plane",
-    "25": "yellow and black fighter plane",
-    "26": "green and black camouflage helicopter",
-    "27": "black helicopter",
-    "28": "red light aircraft",
-    "29": "green and brown camouflage helicopter",
-    "30": "black fighter plane",
-    "31": "grey and yellow fighter plane",
-    "32": "white and black cargo aircraft",
-    "33": "green light aircraft",
-    "34": "white and black fighter plane",
-    "35": "white and black light aircraft",
-    "36": "white drone",
-    "37": "yellow light aircraft",
-    "38": "white and red light aircraft",
-    "39": "yellow and green helicopter",
-    "40": "white and black drone",
-    "41": "white fighter jet",
-    "42": "grey, red, and blue commercial aircraft",
-    "43": "white cargo aircraft",
-    "44": "white, black, and grey missile",
-    "45": "red, white, and blue light aircraft",
-    "46": "red and black drone",
-    "47": "green and white fighter plane",
-    "48": "red fighter plane",
-    "49": "grey commercial aircraft",
-    "50": "grey and red fighter jet",
-    "51": "grey and black fighter plane",
-    "52": "grey camouflage fighter jet",
-    "53": "green camouflage helicopter",
-    "54": "white and red fighter plane",
-    "55": "blue and yellow fighter jet",
-    "56": "blue and white helicopter",
-    "57": "white, black, and red drone",
-    "58": "white and blue cargo aircraft",
-    "59": "green and yellow fighter plane",
-    "60": "red and white helicopter",
-    "61": "grey and green cargo aircraft",
-    "62": "yellow, red, and grey helicopter",
-    "63": "silver fighter plane",
-    "64": "blue, yellow, and green fighter plane",
-    "65": "red and white fighter plane",
-    "66": "grey missile",
-    "67": "white and black fighter jet",
-    "68": "blue missile",
-    "69": "grey light aircraft",
-    "70": "red fighter jet",
-    "71": "yellow missile",
-    "72": "white helicopter",
-    "73": "black drone",
-    "74": "yellow helicopter",
-    "75": "black fighter jet",
-    "76": "white and red helicopter",
-    "77": "grey and white fighter plane",
-    "78": "blue helicopter",
-    "79": "white and blue light aircraft",
-    "80": "grey and black helicopter",
-    "81": "blue and yellow helicopter",
-    "82": "white and blue commercial aircraft",
-    "83": "blue and white missile",
-    "84": "black and brown camouflage helicopter",
-    "85": "red and white fighter jet",
-    "86": "white and orange commercial aircraft",
-    "87": "black and white cargo aircraft",
-    "88": "white and red missile",
-    "89": "white and red commercial aircraft",
-    "90": "white missile",
-    "91": "white and black helicopter",
-    "92": "yellow, red, and blue fighter plane",
-    "93": "red helicopter",
-    "94": "black and white missile",
-    "95": "green helicopter",
-    "96": "green and brown camouflage fighter plane",
-    "97": "grey cargo aircraft",
-    "98": "green and brown camouflage fighter jet",
-    "99": "yellow commercial aircraft",
-    "100": "white and grey helicopter",
-    "101": "white fighter plane",
-    "102": "silver and blue fighter plane",
-    "103": "red and white missile",
-    "104": "green missile",
-    "105": "black and orange drone",
-    "106": "grey fighter plane",
-    "107": "black and yellow missile",
-    "108": "orange light aircraft",
-    "109": "white and blue helicopter",
-    "110": "red and white light aircraft",
-    "111": "grey and red missile",
-    "112": "black and yellow drone",
-    "113": "white and orange light aircraft",
-    "114": "blue, yellow, and white cargo aircraft",
-    "115": "yellow fighter jet",
-    "116": "blue camouflage fighter jet",
-    "117": "green and black missile",
-    "118": "blue, yellow, and black helicopter",
-    "119": "white and red fighter jet",
-    "120": "grey and red commercial aircraft",
-    "121": "red, white, and blue fighter jet",
-    "122": "white, red, and blue commercial aircraft",
-    "123": "blue and white commercial aircraft",
-    "124": "blue and white light aircraft",
-    "125": "red and grey missile"
+    "0": "Coverall",
+    "1": "Face_Shield",
+    "2": "Gloves",
+    "3": "Goggles",
+    "4": "Mask"
   },
   "init_std": 0.02,
   "init_xavier_std": 1.0,
   "is_encoder_decoder": true,
   "label2id": {
-    "black and brown camouflage helicopter": 84,
-    "black and orange drone": 105,
-    "black and white cargo aircraft": 87,
-    "black and white commercial aircraft": 15,
-    "black and white missile": 94,
-    "black and yellow drone": 112,
-    "black and yellow missile": 107,
-    "black camouflage fighter jet": 13,
-    "black cargo aircraft": 17,
-    "black drone": 73,
-    "black fighter jet": 75,
-    "black fighter plane": 30,
-    "black helicopter": 27,
-    "blue and green fighter plane": 7,
-    "blue and grey fighter jet": 21,
-    "blue and red commercial aircraft": 23,
-    "blue and red light aircraft": 14,
-    "blue and white commercial aircraft": 123,
-    "blue and white helicopter": 56,
-    "blue and white light aircraft": 124,
-    "blue and white missile": 83,
-    "blue and yellow fighter jet": 55,
-    "blue and yellow helicopter": 81,
-    "blue camouflage fighter jet": 116,
-    "blue commercial aircraft": 16,
-    "blue helicopter": 78,
-    "blue missile": 68,
-    "blue, yellow, and black helicopter": 118,
-    "blue, yellow, and green fighter plane": 64,
-    "blue, yellow, and white cargo aircraft": 114,
-    "green and black camouflage helicopter": 26,
-    "green and black missile": 117,
-    "green and brown camouflage fighter jet": 98,
-    "green and brown camouflage fighter plane": 96,
-    "green and brown camouflage helicopter": 29,
-    "green and grey helicopter": 22,
-    "green and white fighter plane": 47,
-    "green and yellow fighter plane": 59,
-    "green camouflage helicopter": 53,
-    "green fighter plane": 8,
-    "green helicopter": 95,
-    "green light aircraft": 33,
-    "green missile": 104,
-    "grey and black fighter plane": 51,
-    "grey and black helicopter": 80,
-    "grey and green cargo aircraft": 61,
-    "grey and red commercial aircraft": 120,
-    "grey and red fighter jet": 50,
-    "grey and red missile": 111,
-    "grey and white fighter plane": 77,
-    "grey and white light aircraft": 0,
-    "grey and yellow fighter plane": 31,
-    "grey camouflage fighter jet": 52,
-    "grey cargo aircraft": 97,
-    "grey commercial aircraft": 49,
-    "grey drone": 12,
-    "grey fighter jet": 1,
-    "grey fighter plane": 106,
-    "grey helicopter": 2,
-    "grey light aircraft": 69,
-    "grey missile": 66,
-    "grey, red, and blue commercial aircraft": 42,
-    "orange and black fighter jet": 3,
-    "orange light aircraft": 108,
-    "red and black drone": 46,
-    "red and grey missile": 125,
-    "red and white fighter jet": 85,
-    "red and white fighter plane": 65,
-    "red and white helicopter": 60,
-    "red and white light aircraft": 110,
-    "red and white missile": 103,
-    "red fighter jet": 70,
-    "red fighter plane": 48,
-    "red helicopter": 93,
-    "red light aircraft": 28,
-    "red, white, and blue fighter jet": 121,
-    "red, white, and blue light aircraft": 45,
-    "silver and blue fighter plane": 102,
-    "silver fighter plane": 63,
-    "white and black cargo aircraft": 32,
-    "white and black drone": 40,
-    "white and black fighter jet": 67,
-    "white and black fighter plane": 34,
-    "white and black helicopter": 91,
-    "white and black light aircraft": 35,
-    "white and blue cargo aircraft": 58,
-    "white and blue commercial aircraft": 82,
-    "white and blue fighter jet": 20,
-    "white and blue fighter plane": 10,
-    "white and blue helicopter": 109,
-    "white and blue light aircraft": 79,
-    "white and grey helicopter": 100,
-    "white and orange commercial aircraft": 86,
-    "white and orange light aircraft": 113,
-    "white and red commercial aircraft": 89,
-    "white and red fighter jet": 119,
-    "white and red fighter plane": 54,
-    "white and red helicopter": 76,
-    "white and red light aircraft": 38,
-    "white and red missile": 88,
-    "white and yellow commercial aircraft": 5,
-    "white cargo aircraft": 43,
-    "white commercial aircraft": 4,
-    "white drone": 36,
-    "white fighter jet": 41,
-    "white fighter plane": 101,
-    "white helicopter": 72,
-    "white light aircraft": 6,
-    "white missile": 90,
-    "white, black, and grey missile": 44,
-    "white, black, and red drone": 57,
-    "white, blue, and red commercial aircraft": 9,
-    "white, red, and blue commercial aircraft": 122,
-    "white, red, and green fighter plane": 19,
-    "yellow and black fighter plane": 25,
-    "yellow and green helicopter": 39,
-    "yellow and red light aircraft": 11,
-    "yellow commercial aircraft": 99,
-    "yellow fighter jet": 115,
-    "yellow fighter plane": 24,
-    "yellow helicopter": 74,
-    "yellow light aircraft": 37,
-    "yellow missile": 71,
-    "yellow, black, and red helicopter": 18,
-    "yellow, red, and blue fighter plane": 92,
-    "yellow, red, and grey helicopter": 62
+    "Coverall": 0,
+    "Face_Shield": 1,
+    "Gloves": 2,
+    "Goggles": 3,
+    "Mask": 4
   },
   "mask_loss_coefficient": 1,
   "max_position_embeddings": 1024,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:83d5caf762a55d07c5310d2361c335d9974ada9bfcb1aaf3dfcc2f9ba1ca00a7
-size 166623364
+oid sha256:5754f71bc895f042c7657509b0da23e8cbab2562c9c4d3c29ab52462d9569dac
+size 166498936

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:d937fc3f7cde734668f890761ee2bc00791d827ca40235619f7e54ab25506078
+oid sha256:5eac004492274a59eae205fb9af3d8c8af539e05d13803f3b6aa31ff4e6203dd
 size 5240