cropmodel-tree-genus / config.json
ritesh313's picture
Upload ResNet-18 genus model (epoch 3)
6493d39 verified
{
"cropmodel": {
"architecture": "resnet18",
"batch_size": 4,
"num_workers": 0,
"lr": 0.0001,
"scheduler": {
"type": "ReduceLROnPlateau",
"params": {
"mode": "min",
"factor": 0.5,
"patience": 5,
"threshold": 0.0001,
"threshold_mode": "rel",
"cooldown": 0,
"min_lr": 0,
"eps": 1e-08
}
},
"balance_classes": false,
"resize": [
224,
224
],
"resize_interpolation": "nearest",
"expand": 0,
"label_dict": {
"Abies": 0,
"Acer": 1,
"Alnus": 2,
"Amelanchier": 3,
"Arctostaphylos": 4,
"Betula": 5,
"Bourreria": 6,
"Bucida": 7,
"Bursera": 8,
"Calocedrus": 9,
"Carpinus": 10,
"Carya": 11,
"Celtis": 12,
"Cercis": 13,
"Cornus": 14,
"Diospyros": 15,
"Fagus": 16,
"Fraxinus": 17,
"Gleditsia": 18,
"Gymnocladus": 19,
"Halesia": 20,
"Ilex": 21,
"Juglans": 22,
"Juniperus": 23,
"Larix": 24,
"Liquidambar": 25,
"Liriodendron": 26,
"Maclura": 27,
"Magnolia": 28,
"Metrosideros": 29,
"Morus": 30,
"Nyssa": 31,
"Ostrya": 32,
"Oxydendrum": 33,
"Picea": 34,
"Pinaceae": 35,
"Pinus": 36,
"Pisonia": 37,
"Platanus": 38,
"Populus": 39,
"Prunus": 40,
"Pseudotsuga": 41,
"Quercus": 42,
"Robinia": 43,
"Sassafras": 44,
"Sideroxylon": 45,
"Symphoricarpos": 46,
"Taxus": 47,
"Thuja": 48,
"Tilia": 49,
"Triadica": 50,
"Tsuga": 51,
"Ulmus": 52,
"Unknown": 53
}
}
}