Skip to content

Commit 5cfb2c9

Browse files
Added more tflite models to pretrained tests (#1325)
Signed-off-by: Tom Wildenhain <[email protected]>
1 parent 37d0c28 commit 5cfb2c9

File tree

1 file changed

+49
-1
lines changed

1 file changed

+49
-1
lines changed

tests/run_pretrained_models.yaml

Lines changed: 49 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -468,7 +468,7 @@ deeplabv3_mnv2_ade20k_float_tflite:
468468
outputs:
469469
- ArgMax
470470

471-
deeplabv3_mnv2_ade20k_uint8_tflite:
471+
deeplabv3_mnv2_ade20k_uint8_tflite_dequantize:
472472
tf_min_version: 2.1
473473
disabled: false
474474
url: https://github.com/mlcommons/mobile_models/raw/main/v0_7/tflite/deeplabv3_mnv2_ade20k_uint8.tflite
@@ -481,3 +481,51 @@ deeplabv3_mnv2_ade20k_uint8_tflite:
481481
"MobilenetV2/MobilenetV2/input": [1, 512, 512, 3]
482482
outputs:
483483
- ArgMax
484+
485+
deeplabv3_mnv2_ade20k_uint8_tflite:
486+
tf_min_version: 2.1
487+
disabled: true # Requires ORT nightly for dequantize of int32
488+
url: https://github.com/mlcommons/mobile_models/raw/main/v0_7/tflite/deeplabv3_mnv2_ade20k_uint8.tflite
489+
model: "deeplabv3_mnv2_ade20k_uint8.tflite"
490+
model_type: tflite
491+
input_get: get_ade20k_uint8
492+
opset_constraints:
493+
"onnx":
494+
"min": 10
495+
ptol: 1.5
496+
dequantize: false
497+
inputs:
498+
"MobilenetV2/MobilenetV2/input": [1, 512, 512, 3]
499+
outputs:
500+
- ArgMax
501+
502+
deeplabv3_mnv2_ade20k_int8_tflite_dequantize:
503+
tf_min_version: 2.1
504+
disabled: false
505+
url: https://github.com/mlcommons/mobile_models/raw/main/v0_7/tflite/deeplabv3_mnv2_ade20k_int8.tflite
506+
model: "deeplabv3_mnv2_ade20k_int8.tflite"
507+
model_type: tflite
508+
input_get: get_ade20k_uint8 # the input is uint8 despite the model name
509+
ptol: 3.0
510+
dequantize: true
511+
inputs:
512+
"MobilenetV2/MobilenetV2/input": [1, 512, 512, 3]
513+
outputs:
514+
- ArgMax
515+
516+
deeplabv3_mnv2_ade20k_int8_tflite:
517+
tf_min_version: 2.1
518+
disabled: true # Requires ORT nightly for dequantize of int32
519+
url: https://github.com/mlcommons/mobile_models/raw/main/v0_7/tflite/deeplabv3_mnv2_ade20k_int8.tflite
520+
model: "deeplabv3_mnv2_ade20k_int8.tflite"
521+
model_type: tflite
522+
input_get: get_ade20k_uint8 # the input is uint8 despite the model name
523+
opset_constraints:
524+
"onnx":
525+
"min": 13
526+
ptol: 1.9
527+
dequantize: false
528+
inputs:
529+
"MobilenetV2/MobilenetV2/input": [1, 512, 512, 3]
530+
outputs:
531+
- ArgMax

0 commit comments

Comments
 (0)