Upload ViTForImageClassification

config.json

@@ -1,21 +1,120 @@
 {
-  "_name_or_path": "KhangTruong/NWPU-Net",
   "architectures": [
-    "ViTModel"
+    "ViTForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
+  "dtype": "float32",
+  "encoder_stride": 16,
   "hidden_act": "gelu",
   "hidden_dropout_prob": 0.0,
   "hidden_size": 768,
+  "id2label": {
+    "0": "airplane",
+    "1": "airport",
+    "10": "commercial area",
+    "11": "dense residential",
+    "12": "desert",
+    "13": "forest",
+    "14": "freeway",
+    "15": "golf course",
+    "16": "ground track field",
+    "17": "harbor",
+    "18": "industrial area",
+    "19": "intersection",
+    "2": "baseball diamond",
+    "20": "island",
+    "21": "lake",
+    "22": "meadow",
+    "23": "medium residential",
+    "24": "mobile home park",
+    "25": "mountain",
+    "26": "overpass",
+    "27": "palace",
+    "28": "parking lot",
+    "29": "railway",
+    "3": "basketball court",
+    "30": "railway station",
+    "31": "rectangular farmland",
+    "32": "river",
+    "33": "roundabout",
+    "34": "runway",
+    "35": "sea ice",
+    "36": "ship",
+    "37": "snowberg",
+    "38": "sparse residential",
+    "39": "stadium",
+    "4": "beach",
+    "40": "storage tank",
+    "41": "tennis court",
+    "42": "terrace",
+    "43": "thermal power station",
+    "44": "wetland",
+    "5": "bridge",
+    "6": "chaparral",
+    "7": "church",
+    "8": "circular farmland",
+    "9": "cloud"
+  },
   "image_size": 224,
   "initializer_range": 0.02,
   "intermediate_size": 3072,
+  "label2id": {
+    "airplane": "0",
+    "airport": "1",
+    "baseball diamond": "2",
+    "basketball court": "3",
+    "beach": "4",
+    "bridge": "5",
+    "chaparral": "6",
+    "church": "7",
+    "circular farmland": "8",
+    "cloud": "9",
+    "commercial area": "10",
+    "dense residential": "11",
+    "desert": "12",
+    "forest": "13",
+    "freeway": "14",
+    "golf course": "15",
+    "ground track field": "16",
+    "harbor": "17",
+    "industrial area": "18",
+    "intersection": "19",
+    "island": "20",
+    "lake": "21",
+    "meadow": "22",
+    "medium residential": "23",
+    "mobile home park": "24",
+    "mountain": "25",
+    "overpass": "26",
+    "palace": "27",
+    "parking lot": "28",
+    "railway": "29",
+    "railway station": "30",
+    "rectangular farmland": "31",
+    "river": "32",
+    "roundabout": "33",
+    "runway": "34",
+    "sea ice": "35",
+    "ship": "36",
+    "snowberg": "37",
+    "sparse residential": "38",
+    "stadium": "39",
+    "storage tank": "40",
+    "tennis court": "41",
+    "terrace": "42",
+    "thermal power station": "43",
+    "wetland": "44"
+  },
   "layer_norm_eps": 1e-12,
   "model_type": "vit",
   "num_attention_heads": 12,
   "num_channels": 3,
   "num_hidden_layers": 12,
   "patch_size": 16,
+  "pooler_act": "tanh",
+  "pooler_output_size": 768,
+  "problem_type": "single_label_classification",
   "qkv_bias": true,
-  "transformers_version": "4.13.0.dev0"
-}
+  "transformers_version": "5.0.0",
+  "use_cache": false
+}

model.safetensors

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