diff --git a/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may-int8-quantized.onnx b/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may-int8-quantized.onnx deleted file mode 100644 index cf0dffa22a2fd194df52947e7b3ae2fa1fda4a17..0000000000000000000000000000000000000000 --- a/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may-int8-quantized.onnx +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:933f8332152718d2b7b30ba40cf07fbbb4099d7ecc3709708862e3d36b5661a9 -size 1087947 diff --git a/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may.onnx b/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may.onnx deleted file mode 100644 index ded2d2f25a17a545a0f3dc87a8f2433241b80301..0000000000000000000000000000000000000000 --- a/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may.onnx +++ /dev/null @@ -1,3 +0,0 @@ -version https://git-lfs.github.com/spec/v1 -oid sha256:a383f866bc76140fa609e989749b1c81c2962c1664136d83763b0d8962e10e07 -size 4145769 diff --git a/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar-act_int8-wt_int8-quantized.onnx b/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar-act_int8-wt_int8-quantized.onnx new file mode 100644 index 0000000000000000000000000000000000000000..0fb03c6d7fdfed0ba5f9f9469c5ee389f757ada2 --- /dev/null +++ b/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar-act_int8-wt_int8-quantized.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:c1d64b3d0e5a8470cfba63ea6dc8752188bfdca733836aea22a2310bef836e5c +size 1087530 diff --git a/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar.onnx b/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar.onnx new file mode 100644 index 0000000000000000000000000000000000000000..8e7b5cc8779af42921c9d7b1a4e20f3905cd728c --- /dev/null +++ b/models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar.onnx @@ -0,0 +1,3 @@ +version https://git-lfs.github.com/spec/v1 +oid sha256:6d4978a7b6d25514d5e24811b82bfb511d166bdd8ca3b03aa63c1623d4d039c7 +size 4146213 diff --git a/tools/quantize/quantize-inc.py b/tools/quantize/quantize-inc.py index f6384187a3d92b219b8ba3e94f94b90c96d85992..f72f5c92f5b3df21cbf6528a4e90be26fc08fbfe 100644 --- a/tools/quantize/quantize-inc.py +++ b/tools/quantize/quantize-inc.py @@ -130,9 +130,6 @@ models = dict( mp_handpose=Quantize(model_path='../../models/handpose_estimation_mediapipe/handpose_estimation_mediapipe_2022may.onnx', config_path='./inc_configs/mp_handpose.yaml', custom_dataset=Dataset(root='../../benchmark/data/palm_detection', dim='hwc', swapRB=True, mean=127.5, std=127.5, toFP32=True)), - lpd_yunet=Quantize(model_path='../../models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2022may.onnx', - config_path='./inc_configs/lpd_yunet.yaml', - custom_dataset=Dataset(root='../../benchmark/data/license_plate_detection', size=(320, 240), dim='chw', toFP32=True)), fer=Quantize(model_path='../../models/facial_expression_recognition/facial_expression_recognition_mobilefacenet_2022july.onnx', config_path='./inc_configs/fer.yaml', custom_dataset=FerDataset(root='../../benchmark/data/facial_expression_recognition/fer_calibration', size=(112, 112), toFP32=True, swapRB=True, scale=1./255, mean=0.5, std=0.5), diff --git a/tools/quantize/quantize-ort.py b/tools/quantize/quantize-ort.py index 7a400b1229a9b0212358684010cc8db85e57e641..51a3a0c6e583a3bd7856ae07841ad3b01cc95416 100644 --- a/tools/quantize/quantize-ort.py +++ b/tools/quantize/quantize-ort.py @@ -116,6 +116,9 @@ models=dict( calibration_image_dir='path/to/dataset', transforms=Compose([HandAlign("mp_handpose"), Resize(size=(224, 224)), Normalize(std=[255, 255, 255]), ColorConvert(ctype=cv.COLOR_BGR2RGB)]), data_dim='hwc'), + lpd_yunet=Quantize(model_path='../../models/license_plate_detection_yunet/license_plate_detection_lpd_yunet_2023mar.onnx', + calibration_image_dir='../../benchmark/data/license_plate_detection', + transforms=Compose([Resize(size=(320, 240))])), ) if __name__ == '__main__':