agriscan-efficientnet / config.json
VCDSA's picture
Upload EfficientNetForImageClassification
0800d5c verified
{
"architectures": [
"EfficientNetForImageClassification"
],
"batch_norm_eps": 0.001,
"batch_norm_momentum": 0.99,
"depth_coefficient": 1.0,
"depth_divisor": 8,
"depthwise_padding": [],
"drop_connect_rate": 0.2,
"dropout_rate": 0.2,
"dtype": "float32",
"expand_ratios": [
1,
6,
6,
6,
6,
6,
6
],
"hidden_act": "swish",
"hidden_dim": 1280,
"id2label": {
"0": "Apple scab",
"1": "Apple Black rot",
"2": "Apple Cedar apple rust",
"3": "Apple healthy",
"4": "Blueberry healthy",
"5": "Cherry (including sour) powdery mildew",
"6": "Cherry (including sour) healthy",
"7": "Corn (maize) Cercospora leaf spot gray leaf spot",
"8": "Corn (maize) common rust",
"9": "Corn (maize) northern leaf blight",
"10": "Corn (maize) healthy",
"11": "Grape Black rot",
"12": "Grape Esca (Black Measles)",
"13": "Grape Leaf blight (Isariopsis Leaf Spot)",
"14": "Grape healthy",
"15": "Orange Huanglongbing (greening)",
"16": "Peach Bacterial spot",
"17": "Peach healthy",
"18": "Pepper bell Bacterial spot",
"19": "Pepper bell healthy",
"20": "Potato Early blight",
"21": "Potato Late blight",
"22": "Potato healthy",
"23": "Raspberry healthy",
"24": "Soybean healthy",
"25": "Squash powdery mildew",
"26": "Strawberry Leaf scorch",
"27": "Strawberry healthy",
"28": "Tomato Bacterial spot",
"29": "Tomato Early blight",
"30": "Tomato Late blight",
"31": "Tomato Leaf Mold",
"32": "Tomato Septoria leaf spot",
"33": "Tomato Spider mites Two-spotted spider mite",
"34": "Tomato Target Spot",
"35": "Tomato Tomato yellow leaf curl virus",
"36": "Tomato Tomato mosaic virus",
"37": "Tomato healthy"
},
"image_size": 224,
"in_channels": [
32,
16,
24,
40,
80,
112,
192
],
"initializer_range": 0.02,
"kernel_sizes": [
3,
3,
5,
3,
5,
5,
3
],
"label2id": {
"Apple Black rot": 1,
"Apple Cedar apple rust": 2,
"Apple healthy": 3,
"Apple scab": 0,
"Blueberry healthy": 4,
"Cherry (including sour) healthy": 6,
"Cherry (including sour) powdery mildew": 5,
"Corn (maize) Cercospora leaf spot gray leaf spot": 7,
"Corn (maize) common rust": 8,
"Corn (maize) healthy": 10,
"Corn (maize) northern leaf blight": 9,
"Grape Black rot": 11,
"Grape Esca (Black Measles)": 12,
"Grape Leaf blight (Isariopsis Leaf Spot)": 13,
"Grape healthy": 14,
"Orange Huanglongbing (greening)": 15,
"Peach Bacterial spot": 16,
"Peach healthy": 17,
"Pepper bell Bacterial spot": 18,
"Pepper bell healthy": 19,
"Potato Early blight": 20,
"Potato Late blight": 21,
"Potato healthy": 22,
"Raspberry healthy": 23,
"Soybean healthy": 24,
"Squash powdery mildew": 25,
"Strawberry Leaf scorch": 26,
"Strawberry healthy": 27,
"Tomato Bacterial spot": 28,
"Tomato Early blight": 29,
"Tomato Late blight": 30,
"Tomato Leaf Mold": 31,
"Tomato Septoria leaf spot": 32,
"Tomato Spider mites Two-spotted spider mite": 33,
"Tomato Target Spot": 34,
"Tomato Tomato mosaic virus": 36,
"Tomato Tomato yellow leaf curl virus": 35,
"Tomato healthy": 37
},
"model_type": "efficientnet",
"num_block_repeats": [
1,
2,
2,
3,
3,
4,
1
],
"num_channels": 3,
"num_hidden_layers": 64,
"out_channels": [
16,
24,
40,
80,
112,
192,
320
],
"out_features": null,
"pooling_type": "mean",
"problem_type": "single_label_classification",
"squeeze_expansion_ratio": 0.25,
"stage_names": [
"stem",
"stage1",
"stage2",
"stage3",
"stage4",
"stage5",
"stage6",
"stage7"
],
"strides": [
1,
2,
2,
2,
1,
2,
1
],
"transformers_version": "4.57.3",
"width_coefficient": 1.0
}