mirrors
/
mmyolo
mirror of https://github.com/open-mmlab/mmyolo.git
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[Fix] Fix mmyolo int8 deploy config (#315) 

* fix mmyolo int8 deploy config

* fix lint
pull/286/head
hanrui1sensetime 2022-11-26 09:26:40 +08:00 committed by GitHub
parent 285c395066
commit 15224b7816
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
configs/deploy/detection_tensorrt-int8_dynamic-192x192-960x960.py
View File

@ -10,6 +10,6 @@ backend_config = dict(
min_shape=[1, 3, 192, 192],
opt_shape=[1, 3, 640, 640],
max_shape=[1, 3, 960, 960])))
],
calib_config=dict(create_calib=True, calib_file='calib_data.h5'))
])
calib_config = dict(create_calib=True, calib_file='calib_data.h5')
use_efficientnms = False # whether to replace TRTBatchedNMS plugin with EfficientNMS plugin # noqa E501

4
configs/deploy/detection_tensorrt-int8_static-640x640.py
View File

@ -11,6 +11,6 @@ backend_config = dict(
min_shape=[1, 3, 640, 640],
opt_shape=[1, 3, 640, 640],
max_shape=[1, 3, 640, 640])))
],
calib_config=dict(create_calib=True, calib_file='calib_data.h5'))
])
calib_config = dict(create_calib=True, calib_file='calib_data.h5')
use_efficientnms = False # whether to replace TRTBatchedNMS plugin with EfficientNMS plugin # noqa E501