Upload ViTForImageClassification

README.md

@@ -1,7 +1,7 @@
 ---
-license: mit
 language:
 - en
+license: mit
 ---
 # Logo Recognition Model: a mix of UAE companies and global enterprises
 
@@ -60,14 +60,8 @@ The base model used is the Vision Transformer `vit-base-patch16-224-in21k`, whic
 The model was trained on a curated dataset of UAE company logos as well as others of international companies. The dataset consists of thousands of images across various brands to ensure robustness and accuracy.
 
 ### Performance
-***** eval metrics *****
--  epoch                   =       20.0
--  eval_accuracy           =     0.9761
--  eval_loss               =     0.1193
--  eval_runtime            = 0:00:20.51
--  eval_samples_per_second =    268.951
--  eval_steps_per_second   =      8.432
-  
+The model achieved high accuracy on a held-out validation set, indicating strong performance in classifying UAE company logos. Detailed performance metrics (accuracy, precision, recall, F1-score) can be provided upon request.
+
 ## How to Use
 To use the model for inference, you can load it using the `transformers` library from Hugging Face:
 
@@ -160,7 +154,7 @@ print(model.config.id2label[predicted_label])
 
 
 ### Limitations and Biases
-- The model is specifically trained on UAE company logos and a few from Outside the UAE, it may not perform well on logos from numerous other large or small companies.
+- The model is specifically trained on UAE company logos and may not perform well on logos from companies outside the UAE.
 - The model's performance is contingent upon the quality and diversity of the training dataset.
 - Potential biases in the training data can lead to biases in model predictions.
 

config.json

@@ -1,5 +1,5 @@
 {
-  "_name_or_path": "Falconsai/brand_identification",
+  "_name_or_path": "./model/vit_model/",
   "architectures": [
     "ViTForImageClassification"
   ],
@@ -9,415 +9,135 @@
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
   "id2label": {
-    "0": "acer",
-    "1": "acerr",
+    "0": "abu_dhabi_islamic_bank",
+    "1": "acer",
     "2": "adidas",
-    "3": "aerial7",
-    "4": "agfa",
-    "5": "ainol",
-    "6": "aiptek",
-    "7": "akai",
-    "8": "alienware",
-    "9": "amd",
-    "10": "amkette",
-    "11": "amoi",
-    "12": "amstrad",
-    "13": "andrea_electronics",
-    "14": "apple",
-    "15": "applek",
-    "16": "arm",
-    "17": "asl",
-    "18": "astro",
-    "19": "asus",
-    "20": "asusu",
-    "21": "avaya",
-    "22": "aya",
-    "23": "bang_&_olufsen",
-    "24": "bayern",
-    "25": "bazooka",
-    "26": "bbk",
-    "27": "bcp",
-    "28": "beacon",
-    "29": "beats_by_dre",
-    "30": "blackberry",
-    "31": "blue_star",
-    "32": "bose",
-    "33": "boses",
-    "34": "breaker",
-    "35": "btc",
-    "36": "calistoga",
-    "37": "calumet",
-    "38": "casio",
-    "39": "cce",
-    "40": "cisco_systems",
-    "41": "clarion",
-    "42": "coke",
-    "43": "compact",
-    "44": "connector",
-    "45": "contax",
-    "46": "contec",
-    "47": "contour",
-    "48": "cooke",
-    "49": "currys",
-    "50": "cybertron",
-    "51": "d-link",
-    "52": "daewoo_electronics",
-    "53": "dawa",
-    "54": "dayton",
-    "55": "dell",
-    "56": "delonghi",
-    "57": "denver",
-    "58": "devolo",
-    "59": "digital",
-    "60": "dimension",
-    "61": "ecs",
-    "62": "el_dorado",
-    "63": "emachines",
-    "64": "emirates_airlines",
-    "65": "epson",
-    "66": "fram",
-    "67": "fujifilm",
-    "68": "fujitsu",
-    "69": "gateway",
-    "70": "genius",
-    "71": "goldstar",
-    "72": "google",
-    "73": "gopro",
-    "74": "gorenje",
-    "75": "grundig",
-    "76": "haier",
-    "77": "havells",
-    "78": "hcl",
-    "79": "hec",
-    "80": "hewlett-packard",
-    "81": "high_point",
-    "82": "hisense",
-    "83": "hitachi",
-    "84": "honeywell",
-    "85": "hp",
-    "86": "htc",
-    "87": "huawei",
-    "88": "hughes_&_kettner",
-    "89": "hypson",
-    "90": "ibm",
-    "91": "ifr",
-    "92": "iiyama",
-    "93": "ilford_photo",
-    "94": "intel",
-    "95": "iqinvision",
-    "96": "jenoptik",
-    "97": "jl_audio",
-    "98": "kapsch",
-    "99": "kef",
-    "100": "kenko",
-    "101": "kicker",
-    "102": "kingston",
-    "103": "kodak",
-    "104": "konica_minolta",
-    "105": "leadtek",
-    "106": "leica",
-    "107": "lenovo",
-    "108": "leupold",
-    "109": "lg",
-    "110": "lifesize",
-    "111": "lite-on",
-    "112": "livescribe",
-    "113": "logitech",
-    "114": "louis_vuitton",
-    "115": "luxgen",
-    "116": "magnafon",
-    "117": "mamiya",
-    "118": "man_o'_war",
-    "119": "marshall",
-    "120": "mcdonalds",
-    "121": "mediatek",
-    "122": "medion",
-    "123": "meopta",
-    "124": "mercedes",
-    "125": "merrell",
-    "126": "metz",
-    "127": "microsoft",
-    "128": "minolta",
-    "129": "mobotix",
-    "130": "mophie",
-    "131": "motorola",
-    "132": "msi",
-    "133": "mud_&_bugs",
-    "134": "multitech",
-    "135": "national",
-    "136": "nike",
-    "137": "nikon",
-    "138": "nvidia",
-    "139": "olympus",
-    "140": "oppo",
-    "141": "orbit",
-    "142": "panasonic",
-    "143": "pantech",
-    "144": "pathe",
-    "145": "pentium",
-    "146": "phase_one",
-    "147": "philco",
-    "148": "philips",
-    "149": "pioneer",
-    "150": "planar",
-    "151": "polaroid",
-    "152": "premier",
-    "153": "prima",
-    "154": "proline",
-    "155": "punch",
-    "156": "rbm",
-    "157": "rca",
-    "158": "ricoh",
-    "159": "sagemcom",
-    "160": "sakura",
-    "161": "sampo",
-    "162": "samsung",
-    "163": "san",
-    "164": "sansui",
-    "165": "santa_rosa",
-    "166": "sanyo",
-    "167": "sea_&_sea",
-    "168": "seg",
-    "169": "seitz",
-    "170": "sharp",
-    "171": "sharp_corporation",
-    "172": "sibelius",
-    "173": "sinar",
-    "174": "sinclair",
-    "175": "sony",
-    "176": "sony_ericsson",
-    "177": "spc",
-    "178": "speck",
-    "179": "start",
-    "180": "suzuki",
-    "181": "sva",
-    "182": "tab",
-    "183": "tcl",
-    "184": "technics",
-    "185": "teleavia",
-    "186": "telefunken",
-    "187": "teletech",
-    "188": "thomson",
-    "189": "tma",
-    "190": "tmb",
-    "191": "toshiba",
-    "192": "transcend",
-    "193": "trust",
-    "194": "tunes",
-    "195": "ubuntu",
-    "196": "universum",
-    "197": "vestel",
-    "198": "vim",
-    "199": "vit",
-    "200": "vive",
-    "201": "walton"
+    "3": "adnoc",
+    "4": "aldar",
+    "5": "alienware",
+    "6": "amazon",
+    "7": "amd",
+    "8": "apple",
+    "9": "asus",
+    "10": "beats_by_dre",
+    "11": "blackberry",
+    "12": "bose",
+    "13": "careem",
+    "14": "cisco_systems",
+    "15": "coke",
+    "16": "d-link",
+    "17": "dell",
+    "18": "delonghi",
+    "19": "dp_world",
+    "20": "du",
+    "21": "e&",
+    "22": "emaar",
+    "23": "emirates",
+    "24": "emirates_nbd",
+    "25": "etisalat",
+    "26": "falcons.ai",
+    "27": "first_abu_dhabi_bank",
+    "28": "fujitsu",
+    "29": "google",
+    "30": "gopro",
+    "31": "hec",
+    "32": "hewlett_packard",
+    "33": "hilti",
+    "34": "hisense",
+    "35": "huawei",
+    "36": "ibm",
+    "37": "khaleej_times",
+    "38": "l'or\u00e9al",
+    "39": "lenovo",
+    "40": "lg",
+    "41": "linkedin",
+    "42": "louis_vuitton",
+    "43": "majid_al_futtaim",
+    "44": "mashreq",
+    "45": "maybelline",
+    "46": "mcdonalds",
+    "47": "mercedes",
+    "48": "meta",
+    "49": "microsoft",
+    "50": "msi",
+    "51": "nike",
+    "52": "nvidia",
+    "53": "openai",
+    "54": "puma",
+    "55": "rakez",
+    "56": "samsung",
+    "57": "snapdragon",
+    "58": "tesla",
+    "59": "ubuntu",
+    "60": "virgin",
+    "61": "zwag"
   },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
   "label2id": {
-    "acer": "0",
-    "acerr": "1",
+    "abu_dhabi_islamic_bank": "0",
+    "acer": "1",
     "adidas": "2",
-    "aerial7": "3",
-    "agfa": "4",
-    "ainol": "5",
-    "aiptek": "6",
-    "akai": "7",
-    "alienware": "8",
-    "amd": "9",
-    "amkette": "10",
-    "amoi": "11",
-    "amstrad": "12",
-    "andrea_electronics": "13",
-    "apple": "14",
-    "applek": "15",
-    "arm": "16",
-    "asl": "17",
-    "astro": "18",
-    "asus": "19",
-    "asusu": "20",
-    "avaya": "21",
-    "aya": "22",
-    "bang_&_olufsen": "23",
-    "bayern": "24",
-    "bazooka": "25",
-    "bbk": "26",
-    "bcp": "27",
-    "beacon": "28",
-    "beats_by_dre": "29",
-    "blackberry": "30",
-    "blue_star": "31",
-    "bose": "32",
-    "boses": "33",
-    "breaker": "34",
-    "btc": "35",
-    "calistoga": "36",
-    "calumet": "37",
-    "casio": "38",
-    "cce": "39",
-    "cisco_systems": "40",
-    "clarion": "41",
-    "coke": "42",
-    "compact": "43",
-    "connector": "44",
-    "contax": "45",
-    "contec": "46",
-    "contour": "47",
-    "cooke": "48",
-    "currys": "49",
-    "cybertron": "50",
-    "d-link": "51",
-    "daewoo_electronics": "52",
-    "dawa": "53",
-    "dayton": "54",
-    "dell": "55",
-    "delonghi": "56",
-    "denver": "57",
-    "devolo": "58",
-    "digital": "59",
-    "dimension": "60",
-    "ecs": "61",
-    "el_dorado": "62",
-    "emachines": "63",
-    "emirates_airlines": "64",
-    "epson": "65",
-    "fram": "66",
-    "fujifilm": "67",
-    "fujitsu": "68",
-    "gateway": "69",
-    "genius": "70",
-    "goldstar": "71",
-    "google": "72",
-    "gopro": "73",
-    "gorenje": "74",
-    "grundig": "75",
-    "haier": "76",
-    "havells": "77",
-    "hcl": "78",
-    "hec": "79",
-    "hewlett-packard": "80",
-    "high_point": "81",
-    "hisense": "82",
-    "hitachi": "83",
-    "honeywell": "84",
-    "hp": "85",
-    "htc": "86",
-    "huawei": "87",
-    "hughes_&_kettner": "88",
-    "hypson": "89",
-    "ibm": "90",
-    "ifr": "91",
-    "iiyama": "92",
-    "ilford_photo": "93",
-    "intel": "94",
-    "iqinvision": "95",
-    "jenoptik": "96",
-    "jl_audio": "97",
-    "kapsch": "98",
-    "kef": "99",
-    "kenko": "100",
-    "kicker": "101",
-    "kingston": "102",
-    "kodak": "103",
-    "konica_minolta": "104",
-    "leadtek": "105",
-    "leica": "106",
-    "lenovo": "107",
-    "leupold": "108",
-    "lg": "109",
-    "lifesize": "110",
-    "lite-on": "111",
-    "livescribe": "112",
-    "logitech": "113",
-    "louis_vuitton": "114",
-    "luxgen": "115",
-    "magnafon": "116",
-    "mamiya": "117",
-    "man_o'_war": "118",
-    "marshall": "119",
-    "mcdonalds": "120",
-    "mediatek": "121",
-    "medion": "122",
-    "meopta": "123",
-    "mercedes": "124",
-    "merrell": "125",
-    "metz": "126",
-    "microsoft": "127",
-    "minolta": "128",
-    "mobotix": "129",
-    "mophie": "130",
-    "motorola": "131",
-    "msi": "132",
-    "mud_&_bugs": "133",
-    "multitech": "134",
-    "national": "135",
-    "nike": "136",
-    "nikon": "137",
-    "nvidia": "138",
-    "olympus": "139",
-    "oppo": "140",
-    "orbit": "141",
-    "panasonic": "142",
-    "pantech": "143",
-    "pathe": "144",
-    "pentium": "145",
-    "phase_one": "146",
-    "philco": "147",
-    "philips": "148",
-    "pioneer": "149",
-    "planar": "150",
-    "polaroid": "151",
-    "premier": "152",
-    "prima": "153",
-    "proline": "154",
-    "punch": "155",
-    "rbm": "156",
-    "rca": "157",
-    "ricoh": "158",
-    "sagemcom": "159",
-    "sakura": "160",
-    "sampo": "161",
-    "samsung": "162",
-    "san": "163",
-    "sansui": "164",
-    "santa_rosa": "165",
-    "sanyo": "166",
-    "sea_&_sea": "167",
-    "seg": "168",
-    "seitz": "169",
-    "sharp": "170",
-    "sharp_corporation": "171",
-    "sibelius": "172",
-    "sinar": "173",
-    "sinclair": "174",
-    "sony": "175",
-    "sony_ericsson": "176",
-    "spc": "177",
-    "speck": "178",
-    "start": "179",
-    "suzuki": "180",
-    "sva": "181",
-    "tab": "182",
-    "tcl": "183",
-    "technics": "184",
-    "teleavia": "185",
-    "telefunken": "186",
-    "teletech": "187",
-    "thomson": "188",
-    "tma": "189",
-    "tmb": "190",
-    "toshiba": "191",
-    "transcend": "192",
-    "trust": "193",
-    "tunes": "194",
-    "ubuntu": "195",
-    "universum": "196",
-    "vestel": "197",
-    "vim": "198",
-    "vit": "199",
-    "vive": "200",
-    "walton": "201"
+    "adnoc": "3",
+    "aldar": "4",
+    "alienware": "5",
+    "amazon": "6",
+    "amd": "7",
+    "apple": "8",
+    "asus": "9",
+    "beats_by_dre": "10",
+    "blackberry": "11",
+    "bose": "12",
+    "careem": "13",
+    "cisco_systems": "14",
+    "coke": "15",
+    "d-link": "16",
+    "dell": "17",
+    "delonghi": "18",
+    "dp_world": "19",
+    "du": "20",
+    "e&": "21",
+    "emaar": "22",
+    "emirates": "23",
+    "emirates_nbd": "24",
+    "etisalat": "25",
+    "falcons.ai": "26",
+    "first_abu_dhabi_bank": "27",
+    "fujitsu": "28",
+    "google": "29",
+    "gopro": "30",
+    "hec": "31",
+    "hewlett_packard": "32",
+    "hilti": "33",
+    "hisense": "34",
+    "huawei": "35",
+    "ibm": "36",
+    "khaleej_times": "37",
+    "l'or\u00e9al": "38",
+    "lenovo": "39",
+    "lg": "40",
+    "linkedin": "41",
+    "louis_vuitton": "42",
+    "majid_al_futtaim": "43",
+    "mashreq": "44",
+    "maybelline": "45",
+    "mcdonalds": "46",
+    "mercedes": "47",
+    "meta": "48",
+    "microsoft": "49",
+    "msi": "50",
+    "nike": "51",
+    "nvidia": "52",
+    "openai": "53",
+    "puma": "54",
+    "rakez": "55",
+    "samsung": "56",
+    "snapdragon": "57",
+    "tesla": "58",
+    "ubuntu": "59",
+    "virgin": "60",
+    "zwag": "61"
   },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:7b5ee0a18e998068c3d460e7d60d3df325a26c30677f6769a8c6c491a1341511
-size 343839184
+oid sha256:ba0f32b0045483281c4de1682737579359be7eb4664f0829eb1540cb61451bc9
+size 343408544