BSICLE / mobilevitv2 /inference_config.json
lterriel's picture
Upload 44 files
6815986 verified
{
"backend": "transformers",
"model_name": "apple/mobilevitv2-1.0-imagenet1k-256",
"img_size": 288,
"mean": [
0.485,
0.456,
0.406
],
"std": [
0.229,
0.224,
0.225
],
"labels": [
"0",
"1"
],
"threshold": 0.91166090965271,
"input_name": "pixel_values",
"output_name": "logits",
"val_metrics": {
"ap": 0.9998433583959898,
"auc": 0.9998391248391248,
"threshold": 0.91166090965271,
"per_label": {
"report_by_label": {
"0": {
"precision": 0.9910714285714286,
"recall": 1.0,
"f1-score": 0.9955156950672646,
"support": 111.0
},
"1": {
"precision": 1.0,
"recall": 0.9910714285714286,
"f1-score": 0.9955156950672646,
"support": 112.0
}
},
"accuracy": 0.9955156950672646,
"confusion_matrix": {
"labels": [
"0",
"1"
],
"matrix": [
[
111,
0
],
[
1,
111
]
]
}
},
"threshold_best_f1_score": 0.9955156950667645
},
"test_metrics": {
"ap": 0.9991997793851138,
"auc": 0.9991956241956241,
"threshold": 0.91166090965271,
"per_label": {
"report_by_label": {
"0": {
"precision": 0.9572649572649573,
"recall": 1.0,
"f1-score": 0.9781659388646288,
"support": 112.0
},
"1": {
"precision": 1.0,
"recall": 0.954954954954955,
"f1-score": 0.9769585253456221,
"support": 111.0
}
},
"accuracy": 0.9775784753363229,
"confusion_matrix": {
"labels": [
"0",
"1"
],
"matrix": [
[
112,
0
],
[
5,
106
]
]
}
}
}
}