| { | |
| "_name_or_path": "microsoft/resnet-50", | |
| "architectures": [ | |
| "ResNetForImageClassification" | |
| ], | |
| "depths": [ | |
| 3, | |
| 4, | |
| 6, | |
| 3 | |
| ], | |
| "downsample_in_bottleneck": false, | |
| "downsample_in_first_stage": false, | |
| "embedding_size": 64, | |
| "hidden_act": "relu", | |
| "hidden_sizes": [ | |
| 256, | |
| 512, | |
| 1024, | |
| 2048 | |
| ], | |
| "id2label": { | |
| "0": "apple", | |
| "1": "banana", | |
| "10": "cucumber", | |
| "11": "eggplant", | |
| "12": "garlic", | |
| "13": "ginger", | |
| "14": "grapes", | |
| "15": "jalepeno", | |
| "16": "kiwi", | |
| "17": "lemon", | |
| "18": "lettuce", | |
| "19": "mango", | |
| "2": "beetroot", | |
| "20": "onion", | |
| "21": "orange", | |
| "22": "paprika", | |
| "23": "pear", | |
| "24": "peas", | |
| "25": "pineapple", | |
| "26": "pomegranate", | |
| "27": "potato", | |
| "28": "raddish", | |
| "29": "soy beans", | |
| "3": "bell pepper", | |
| "30": "spinach", | |
| "31": "sweetcorn", | |
| "32": "sweetpotato", | |
| "33": "tomato", | |
| "34": "turnip", | |
| "35": "watermelon", | |
| "4": "cabbage", | |
| "5": "capsicum", | |
| "6": "carrot", | |
| "7": "cauliflower", | |
| "8": "chilli pepper", | |
| "9": "corn" | |
| }, | |
| "label2id": { | |
| "apple": "0", | |
| "banana": "1", | |
| "beetroot": "2", | |
| "bell pepper": "3", | |
| "cabbage": "4", | |
| "capsicum": "5", | |
| "carrot": "6", | |
| "cauliflower": "7", | |
| "chilli pepper": "8", | |
| "corn": "9", | |
| "cucumber": "10", | |
| "eggplant": "11", | |
| "garlic": "12", | |
| "ginger": "13", | |
| "grapes": "14", | |
| "jalepeno": "15", | |
| "kiwi": "16", | |
| "lemon": "17", | |
| "lettuce": "18", | |
| "mango": "19", | |
| "onion": "20", | |
| "orange": "21", | |
| "paprika": "22", | |
| "pear": "23", | |
| "peas": "24", | |
| "pineapple": "25", | |
| "pomegranate": "26", | |
| "potato": "27", | |
| "raddish": "28", | |
| "soy beans": "29", | |
| "spinach": "30", | |
| "sweetcorn": "31", | |
| "sweetpotato": "32", | |
| "tomato": "33", | |
| "turnip": "34", | |
| "watermelon": "35" | |
| }, | |
| "layer_type": "bottleneck", | |
| "model_type": "resnet", | |
| "num_channels": 3, | |
| "out_features": [ | |
| "stage4" | |
| ], | |
| "out_indices": [ | |
| 4 | |
| ], | |
| "problem_type": "single_label_classification", | |
| "stage_names": [ | |
| "stem", | |
| "stage1", | |
| "stage2", | |
| "stage3", | |
| "stage4" | |
| ], | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.41.2" | |
| } | |