chart-recognizer / config.json
StephanAkkerman's picture
Add model
a290f88 verified
{
"architecture": "efficientnet_b0",
"num_classes": 2,
"num_features": 1280,
"label_names": [
"chart",
"non-chart"
],
"pretrained_cfg": {
"tag": "ra_in1k",
"custom_load": false,
"input_size": [
3,
224,
224
],
"fixed_input_size": false,
"interpolation": "bicubic",
"crop_pct": 0.875,
"crop_mode": "center",
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"num_classes": 1000,
"pool_size": [
7,
7
],
"first_conv": "conv_stem",
"classifier": "classifier"
}
}