image-classifier-efficientnet / model_config.json
nailarais1's picture
Update model_config.json
4091c22 verified
Raw
History Blame Contribute Delete
5.51 kB
{
"_name_or_path": "nailarais1/image-classifier-efficientnet",
"architectures": [
"EfficientNet"
],
"num_classes": 102,
"model_type": "cnn",
"task": "image-classification",
"image_size": 224,
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"training_epochs": 3,
"dataset": "oxford_flowers102",
"dataset_source": "dpdl-benchmark/oxford_flowers102",
"id2label": {
"0": "pink primrose",
"1": "globe thistle",
"2": "blanket flower",
"3": "trumpet creeper",
"4": "blackberry lily",
"5": "snapdragon",
"6": "colt's foot",
"7": "king protea",
"8": "spear thistle",
"9": "yellow iris",
"10": "globe-flower",
"11": "purple coneflower",
"12": "peruvian lily",
"13": "balloon flower",
"14": "hard-leaved pocket orchid",
"15": "giant white arum lily",
"16": "fire lily",
"17": "pincushion flower",
"18": "fritillary",
"19": "red ginger",
"20": "grape hyacinth",
"21": "corn poppy",
"22": "prince of wales feathers",
"23": "stemless gentian",
"24": "artichoke",
"25": "canterbury bells",
"26": "sweet william",
"27": "carnation",
"28": "garden phlox",
"29": "love in the mist",
"30": "mexican aster",
"31": "alpine sea holly",
"32": "ruby-lipped cattleya",
"33": "cape flower",
"34": "great masterwort",
"35": "siam tulip",
"36": "sweet pea",
"37": "lenten rose",
"38": "barbeton daisy",
"39": "daffodil",
"40": "sword lily",
"41": "poinsettia",
"42": "bolero deep blue",
"43": "wallflower",
"44": "marigold",
"45": "buttercup",
"46": "oxeye daisy",
"47": "english marigold",
"48": "common dandelion",
"49": "petunia",
"50": "wild pansy",
"51": "primula",
"52": "sunflower",
"53": "pelargonium",
"54": "bishop of llandaff",
"55": "gaura",
"56": "geranium",
"57": "orange dahlia",
"58": "tiger lily",
"59": "pink-yellow dahlia",
"60": "cautleya spicata",
"61": "japanese anemone",
"62": "black-eyed susan",
"63": "silverbush",
"64": "californian poppy",
"65": "osteospermum",
"66": "spring crocus",
"67": "bearded iris",
"68": "windflower",
"69": "moon orchid",
"70": "tree poppy",
"71": "gazania",
"72": "azalea",
"73": "water lily",
"74": "rose",
"75": "thorn apple",
"76": "morning glory",
"77": "passion flower",
"78": "lotus lotus",
"79": "toad lily",
"80": "bird of paradise",
"81": "anthurium",
"82": "frangipani",
"83": "clematis",
"84": "hibiscus",
"85": "columbine",
"86": "desert-rose",
"87": "tree mallow",
"88": "magnolia",
"89": "cyclamen",
"90": "watercress",
"91": "monkshood",
"92": "canna lily",
"93": "hippeastrum",
"94": "bee balm",
"95": "ball moss",
"96": "foxglove",
"97": "bougainvillea",
"98": "camellia",
"99": "mallow",
"100": "mexican petunia",
"101": "bromelia"
},
"label2id": {
"pink primrose": 0,
"globe thistle": 1,
"blanket flower": 2,
"trumpet creeper": 3,
"blackberry lily": 4,
"snapdragon": 5,
"colt's foot": 6,
"king protea": 7,
"spear thistle": 8,
"yellow iris": 9,
"globe-flower": 10,
"purple coneflower": 11,
"peruvian lily": 12,
"balloon flower": 13,
"hard-leaved pocket orchid": 14,
"giant white arum lily": 15,
"fire lily": 16,
"pincushion flower": 17,
"fritillary": 18,
"red ginger": 19,
"grape hyacinth": 20,
"corn poppy": 21,
"prince of wales feathers": 22,
"stemless gentian": 23,
"artichoke": 24,
"canterbury bells": 25,
"sweet william": 26,
"carnation": 27,
"garden phlox": 28,
"love in the mist": 29,
"mexican aster": 30,
"alpine sea holly": 31,
"ruby-lipped cattleya": 32,
"cape flower": 33,
"great masterwort": 34,
"siam tulip": 35,
"sweet pea": 36,
"lenten rose": 37,
"barbeton daisy": 38,
"daffodil": 39,
"sword lily": 40,
"poinsettia": 41,
"bolero deep blue": 42,
"wallflower": 43,
"marigold": 44,
"buttercup": 45,
"oxeye daisy": 46,
"english marigold": 47,
"common dandelion": 48,
"petunia": 49,
"wild pansy": 50,
"primula": 51,
"sunflower": 52,
"pelargonium": 53,
"bishop of llandaff": 54,
"gaura": 55,
"geranium": 56,
"orange dahlia": 57,
"tiger lily": 58,
"pink-yellow dahlia": 59,
"cautleya spicata": 60,
"japanese anemone": 61,
"black-eyed susan": 62,
"silverbush": 63,
"californian poppy": 64,
"osteospermum": 65,
"spring crocus": 66,
"bearded iris": 67,
"windflower": 68,
"moon orchid": 69,
"tree poppy": 70,
"gazania": 71,
"azalea": 72,
"water lily": 73,
"rose": 74,
"thorn apple": 75,
"morning glory": 76,
"passion flower": 77,
"lotus lotus": 78,
"toad lily": 79,
"bird of paradise": 80,
"anthurium": 81,
"frangipani": 82,
"clematis": 83,
"hibiscus": 84,
"columbine": 85,
"desert-rose": 86,
"tree mallow": 87,
"magnolia": 88,
"cyclamen": 89,
"watercress": 90,
"monkshood": 91,
"canna lily": 92,
"hippeastrum": 93,
"bee balm": 94,
"ball moss": 95,
"foxglove": 96,
"bougainvillea": 97,
"camellia": 98,
"mallow": 99,
"mexican petunia": 100,
"bromelia": 101
}
}