Upload folder using huggingface_hub

README.md

@@ -0,0 +1,69 @@
+---
+language: en
+license: apache-2.0
+library_name: transformers
+tags:
+- image-classification
+- convnextv2
+- pytorch
+datasets:
+- custom
+metrics:
+- accuracy
+- f1
+---
+
+# huytranduck/convnextv2_50x_dataset
+
+## Model Description
+
+This is a fine-tuned ConvNeXtV2-Atto model for image classification with 100 classes.
+
+### Classes
+['Acacia melanoxylon', 'Acer saccharinum', 'Afzelia africana', 'Afzelia pachyloba', 'Afzelia quanzensis', 'Albizia lucida (Albizia lucidior)', 'Allophylus cobbe (Pometia pinnata)', 'Anisoptera costata (Anisoptera Robusta)', 'Apuleia\xa0leiocarpa', 'Artocarpus calophyllus (Artocarpus asperulus)', 'Artocarpus heterophyllus', 'Autranella congolensis', 'Berlinia bracteosa', 'Betula pendula', 'Bobgunnia fistuloides (Swartzia fistuloides)', 'Brachystegia sp', 'Burckella obovata', 'Burretiodendron tonkinense', 'Callitris columellaris', 'Calocedrus sp', 'Canarium album', 'Chrysophyllum sp', 'Cinnamomum camphora', 'Clarisia racemosa', 'Colophospermum mopane', 'Cunninghamia lanceolata', 'Cupressus funebris (Cupressus pendula)', 'Cylicodiscus gabunensis', 'Dalbergia cochinchinensis', 'Dalbergia oliveri', 'Detarium macrocarpum', 'Dialium bipindense', 'Didelotia africana', 'Diospyros mun', 'Diospyros salletii', 'Distemonanthus benthamianus', 'Engelhardia chrysolepis (Engelhardia roxburghiana)', 'Entandrophragma cylindricum', 'Entandrophragma utile', 'Erythrophleum fordii\xa0', 'Erythrophleum ivorense', 'Eucalyptus cladocalyx', 'Eucalyptus grandis', 'Eucalyptus microcorys', 'Eucalyptus saligna', 'Fokienia hodginsii', 'Fraxinus excelsior', 'Gilbertiodendron dewevrei', 'Guarea cedrata', 'Guibourtia coleosperma', 'Heritiera littoralis', 'Hevea brasiliensis', 'Homalium caryophyllaceum', 'Homalium foetidum', 'Hopea iriana', 'Hopea pierrei', 'Hymenaea courbaril', 'Hymenolobium heterocarpum', 'Juglans regia', 'Khaya senegalensis', 'Klainedoxa gabonensis', 'Lithocarpus ducampii', 'Lophira alata', 'Magnolia hypolampra', 'Martiodendron parviflorum', 'Milicia excelsa', 'Milicia regia', 'Millettia laurentii', 'Monopetalanthus letestui (Bikinia letestui)', 'Myracrodruon urundeuva', 'Myroxylon balsamum', 'Myroxylon balsamum_v2', 'Myroxylon peruiferum', 'Nauclea diderrichii', 'Pachyelasma tessmannii', 'Palaquium waburgianum', 'Pericopsis elata', 'Pinus sp', 'Piptadeniastrum africanum', 'Populus sp', 'Prunus serotina', 'Pterocarpus macrocarpus', 'Pterocarpus soyauxii', 'Pterocarpus sp', 'Qualea paraensis', 'Quercus petraea', 'Quercus robur', 'Quercus rubra', 'Samanea saman', 'Shorea hypochra (Anthoshorea hypochra)', 'Shorea roxburghii (Anthoshorea roxburghii)', 'Sindora cochinchinensis', 'Staudtia stipitata', 'Syzygium hemisphericum (Syzygium chanlos)', 'Tarrietia cochinchinensis (Heritiera cochinchinesis)', 'Tectona grandis', 'Terminalia superba', 'Tetraberlinia bifoliolata', 'Toona sureni', 'Xylia xylocarpa']
+
+## Training Details
+
+- Base model: facebook/convnextv2-atto-1k-224
+- Fine-tuned on custom dataset
+- Number of classes: 100
+- Image size: 224x224
+
+### Training Configuration
+- Epochs: 2 (as specified in your code)
+- Batch size: 64
+- Learning rate: 1e-4
+- Optimizer: AdamW
+- Loss function: CrossEntropyLoss
+
+### Performance Metrics
+- Metrics will be updated after evaluation
+
+
+## Usage
+
+```python
+from transformers import ConvNextV2ForImageClassification, AutoImageProcessor
+import torch
+from PIL import Image
+
+# Load model and processor
+model = ConvNextV2ForImageClassification.from_pretained("huytranduck/convnextv2_50x_dataset")
+processor = AutoImageProcessor.from_pretrained("huytranduck/convnextv2_50x_dataset")
+
+# Load and preprocess image
+image = Image.open("path_to_your_image.jpg")
+inputs = processor(image, return_tensors="pt")
+
+# Make prediction
+with torch.no_grad():
+    outputs = model(**inputs)
+    predictions = torch.nn.functional.softmax(outputs.logits, dim=-1)
+    predicted_class = torch.argmax(predictions, dim=-1)
+
+print(f"Predicted class: {predicted_class.item()}")
+```
+
+## Training Code
+
+The model was trained using PyTorch and Transformers library with custom data augmentation.

config.json

@@ -0,0 +1,245 @@
+{
+  "architectures": [
+    "ConvNextV2ForImageClassification"
+  ],
+  "depths": [
+    2,
+    2,
+    6,
+    2
+  ],
+  "drop_path_rate": 0.0,
+  "hidden_act": "gelu",
+  "hidden_sizes": [
+    40,
+    80,
+    160,
+    320
+  ],
+  "id2label": {
+    "0": "LABEL_0",
+    "1": "LABEL_1",
+    "2": "LABEL_2",
+    "3": "LABEL_3",
+    "4": "LABEL_4",
+    "5": "LABEL_5",
+    "6": "LABEL_6",
+    "7": "LABEL_7",
+    "8": "LABEL_8",
+    "9": "LABEL_9",
+    "10": "LABEL_10",
+    "11": "LABEL_11",
+    "12": "LABEL_12",
+    "13": "LABEL_13",
+    "14": "LABEL_14",
+    "15": "LABEL_15",
+    "16": "LABEL_16",
+    "17": "LABEL_17",
+    "18": "LABEL_18",
+    "19": "LABEL_19",
+    "20": "LABEL_20",
+    "21": "LABEL_21",
+    "22": "LABEL_22",
+    "23": "LABEL_23",
+    "24": "LABEL_24",
+    "25": "LABEL_25",
+    "26": "LABEL_26",
+    "27": "LABEL_27",
+    "28": "LABEL_28",
+    "29": "LABEL_29",
+    "30": "LABEL_30",
+    "31": "LABEL_31",
+    "32": "LABEL_32",
+    "33": "LABEL_33",
+    "34": "LABEL_34",
+    "35": "LABEL_35",
+    "36": "LABEL_36",
+    "37": "LABEL_37",
+    "38": "LABEL_38",
+    "39": "LABEL_39",
+    "40": "LABEL_40",
+    "41": "LABEL_41",
+    "42": "LABEL_42",
+    "43": "LABEL_43",
+    "44": "LABEL_44",
+    "45": "LABEL_45",
+    "46": "LABEL_46",
+    "47": "LABEL_47",
+    "48": "LABEL_48",
+    "49": "LABEL_49",
+    "50": "LABEL_50",
+    "51": "LABEL_51",
+    "52": "LABEL_52",
+    "53": "LABEL_53",
+    "54": "LABEL_54",
+    "55": "LABEL_55",
+    "56": "LABEL_56",
+    "57": "LABEL_57",
+    "58": "LABEL_58",
+    "59": "LABEL_59",
+    "60": "LABEL_60",
+    "61": "LABEL_61",
+    "62": "LABEL_62",
+    "63": "LABEL_63",
+    "64": "LABEL_64",
+    "65": "LABEL_65",
+    "66": "LABEL_66",
+    "67": "LABEL_67",
+    "68": "LABEL_68",
+    "69": "LABEL_69",
+    "70": "LABEL_70",
+    "71": "LABEL_71",
+    "72": "LABEL_72",
+    "73": "LABEL_73",
+    "74": "LABEL_74",
+    "75": "LABEL_75",
+    "76": "LABEL_76",
+    "77": "LABEL_77",
+    "78": "LABEL_78",
+    "79": "LABEL_79",
+    "80": "LABEL_80",
+    "81": "LABEL_81",
+    "82": "LABEL_82",
+    "83": "LABEL_83",
+    "84": "LABEL_84",
+    "85": "LABEL_85",
+    "86": "LABEL_86",
+    "87": "LABEL_87",
+    "88": "LABEL_88",
+    "89": "LABEL_89",
+    "90": "LABEL_90",
+    "91": "LABEL_91",
+    "92": "LABEL_92",
+    "93": "LABEL_93",
+    "94": "LABEL_94",
+    "95": "LABEL_95",
+    "96": "LABEL_96",
+    "97": "LABEL_97",
+    "98": "LABEL_98",
+    "99": "LABEL_99"
+  },
+  "image_size": 224,
+  "initializer_range": 0.02,
+  "label2id": {
+    "LABEL_0": 0,
+    "LABEL_1": 1,
+    "LABEL_10": 10,
+    "LABEL_11": 11,
+    "LABEL_12": 12,
+    "LABEL_13": 13,
+    "LABEL_14": 14,
+    "LABEL_15": 15,
+    "LABEL_16": 16,
+    "LABEL_17": 17,
+    "LABEL_18": 18,
+    "LABEL_19": 19,
+    "LABEL_2": 2,
+    "LABEL_20": 20,
+    "LABEL_21": 21,
+    "LABEL_22": 22,
+    "LABEL_23": 23,
+    "LABEL_24": 24,
+    "LABEL_25": 25,
+    "LABEL_26": 26,
+    "LABEL_27": 27,
+    "LABEL_28": 28,
+    "LABEL_29": 29,
+    "LABEL_3": 3,
+    "LABEL_30": 30,
+    "LABEL_31": 31,
+    "LABEL_32": 32,
+    "LABEL_33": 33,
+    "LABEL_34": 34,
+    "LABEL_35": 35,
+    "LABEL_36": 36,
+    "LABEL_37": 37,
+    "LABEL_38": 38,
+    "LABEL_39": 39,
+    "LABEL_4": 4,
+    "LABEL_40": 40,
+    "LABEL_41": 41,
+    "LABEL_42": 42,
+    "LABEL_43": 43,
+    "LABEL_44": 44,
+    "LABEL_45": 45,
+    "LABEL_46": 46,
+    "LABEL_47": 47,
+    "LABEL_48": 48,
+    "LABEL_49": 49,
+    "LABEL_5": 5,
+    "LABEL_50": 50,
+    "LABEL_51": 51,
+    "LABEL_52": 52,
+    "LABEL_53": 53,
+    "LABEL_54": 54,
+    "LABEL_55": 55,
+    "LABEL_56": 56,
+    "LABEL_57": 57,
+    "LABEL_58": 58,
+    "LABEL_59": 59,
+    "LABEL_6": 6,
+    "LABEL_60": 60,
+    "LABEL_61": 61,
+    "LABEL_62": 62,
+    "LABEL_63": 63,
+    "LABEL_64": 64,
+    "LABEL_65": 65,
+    "LABEL_66": 66,
+    "LABEL_67": 67,
+    "LABEL_68": 68,
+    "LABEL_69": 69,
+    "LABEL_7": 7,
+    "LABEL_70": 70,
+    "LABEL_71": 71,
+    "LABEL_72": 72,
+    "LABEL_73": 73,
+    "LABEL_74": 74,
+    "LABEL_75": 75,
+    "LABEL_76": 76,
+    "LABEL_77": 77,
+    "LABEL_78": 78,
+    "LABEL_79": 79,
+    "LABEL_8": 8,
+    "LABEL_80": 80,
+    "LABEL_81": 81,
+    "LABEL_82": 82,
+    "LABEL_83": 83,
+    "LABEL_84": 84,
+    "LABEL_85": 85,
+    "LABEL_86": 86,
+    "LABEL_87": 87,
+    "LABEL_88": 88,
+    "LABEL_89": 89,
+    "LABEL_9": 9,
+    "LABEL_90": 90,
+    "LABEL_91": 91,
+    "LABEL_92": 92,
+    "LABEL_93": 93,
+    "LABEL_94": 94,
+    "LABEL_95": 95,
+    "LABEL_96": 96,
+    "LABEL_97": 97,
+    "LABEL_98": 98,
+    "LABEL_99": 99
+  },
+  "layer_norm_eps": 1e-12,
+  "model_type": "convnextv2",
+  "num_channels": 3,
+  "num_stages": 4,
+  "out_features": [
+    "stage4"
+  ],
+  "out_indices": [
+    4
+  ],
+  "patch_size": 4,
+  "stage_names": [
+    "stem",
+    "stage1",
+    "stage2",
+    "stage3",
+    "stage4"
+  ],
+  "torch_dtype": "float32",
+  "transformers_version": "4.52.4"
+}

model.safetensors

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

model_info.json

@@ -0,0 +1,108 @@
+{
+  "num_classes": 100,
+  "class_names": [
+    "Acacia melanoxylon",
+    "Acer saccharinum",
+    "Afzelia africana",
+    "Afzelia pachyloba",
+    "Afzelia quanzensis",
+    "Albizia lucida (Albizia lucidior)",
+    "Allophylus cobbe (Pometia pinnata)",
+    "Anisoptera costata (Anisoptera Robusta)",
+    "Apuleia\u00a0leiocarpa",
+    "Artocarpus calophyllus (Artocarpus asperulus)",
+    "Artocarpus heterophyllus",
+    "Autranella congolensis",
+    "Berlinia bracteosa",
+    "Betula pendula",
+    "Bobgunnia fistuloides (Swartzia fistuloides)",
+    "Brachystegia sp",
+    "Burckella obovata",
+    "Burretiodendron tonkinense",
+    "Callitris columellaris",
+    "Calocedrus sp",
+    "Canarium album",
+    "Chrysophyllum sp",
+    "Cinnamomum camphora",
+    "Clarisia racemosa",
+    "Colophospermum mopane",
+    "Cunninghamia lanceolata",
+    "Cupressus funebris (Cupressus pendula)",
+    "Cylicodiscus gabunensis",
+    "Dalbergia cochinchinensis",
+    "Dalbergia oliveri",
+    "Detarium macrocarpum",
+    "Dialium bipindense",
+    "Didelotia africana",
+    "Diospyros mun",
+    "Diospyros salletii",
+    "Distemonanthus benthamianus",
+    "Engelhardia chrysolepis (Engelhardia roxburghiana)",
+    "Entandrophragma cylindricum",
+    "Entandrophragma utile",
+    "Erythrophleum fordii\u00a0",
+    "Erythrophleum ivorense",
+    "Eucalyptus cladocalyx",
+    "Eucalyptus grandis",
+    "Eucalyptus microcorys",
+    "Eucalyptus saligna",
+    "Fokienia hodginsii",
+    "Fraxinus excelsior",
+    "Gilbertiodendron dewevrei",
+    "Guarea cedrata",
+    "Guibourtia coleosperma",
+    "Heritiera littoralis",
+    "Hevea brasiliensis",
+    "Homalium caryophyllaceum",
+    "Homalium foetidum",
+    "Hopea iriana",
+    "Hopea pierrei",
+    "Hymenaea courbaril",
+    "Hymenolobium heterocarpum",
+    "Juglans regia",
+    "Khaya senegalensis",
+    "Klainedoxa gabonensis",
+    "Lithocarpus ducampii",
+    "Lophira alata",
+    "Magnolia hypolampra",
+    "Martiodendron parviflorum",
+    "Milicia excelsa",
+    "Milicia regia",
+    "Millettia laurentii",
+    "Monopetalanthus letestui (Bikinia letestui)",
+    "Myracrodruon urundeuva",
+    "Myroxylon balsamum",
+    "Myroxylon balsamum_v2",
+    "Myroxylon peruiferum",
+    "Nauclea diderrichii",
+    "Pachyelasma tessmannii",
+    "Palaquium waburgianum",
+    "Pericopsis elata",
+    "Pinus sp",
+    "Piptadeniastrum africanum",
+    "Populus sp",
+    "Prunus serotina",
+    "Pterocarpus macrocarpus",
+    "Pterocarpus soyauxii",
+    "Pterocarpus sp",
+    "Qualea paraensis",
+    "Quercus petraea",
+    "Quercus robur",
+    "Quercus rubra",
+    "Samanea saman",
+    "Shorea hypochra (Anthoshorea hypochra)",
+    "Shorea roxburghii (Anthoshorea roxburghii)",
+    "Sindora cochinchinensis",
+    "Staudtia stipitata",
+    "Syzygium hemisphericum (Syzygium chanlos)",
+    "Tarrietia cochinchinensis (Heritiera cochinchinesis)",
+    "Tectona grandis",
+    "Terminalia superba",
+    "Tetraberlinia bifoliolata",
+    "Toona sureni",
+    "Xylia xylocarpa"
+  ],
+  "image_size": 224,
+  "model_type": "convnextv2-atto",
+  "metrics": {}
+}

preprocessor_config.json

@@ -0,0 +1,22 @@
+{
+  "crop_pct": 0.875,
+  "do_normalize": true,
+  "do_rescale": true,
+  "do_resize": true,
+  "image_mean": [
+    0.485,
+    0.456,
+    0.406
+  ],
+  "image_processor_type": "ConvNextImageProcessor",
+  "image_std": [
+    0.229,
+    0.224,
+    0.225
+  ],
+  "resample": 3,
+  "rescale_factor": 0.00392156862745098,
+  "size": {
+    "shortest_edge": 224
+  }
+}