fix reg
parent
c5edb85550
commit
fe56bffb5b
|
@ -103,7 +103,7 @@ tensorrt:
|
||||||
pipeline_seg_trt_dynamic_fp32: &pipeline_seg_trt_dynamic_fp32
|
pipeline_seg_trt_dynamic_fp32: &pipeline_seg_trt_dynamic_fp32
|
||||||
convert_image: *convert_image
|
convert_image: *convert_image
|
||||||
backend_test: *default_backend_test
|
backend_test: *default_backend_test
|
||||||
# sdk_config: *sdk_seg_dynamic
|
sdk_config: *sdk_seg_dynamic
|
||||||
deploy_config: configs/mmdet/instance-seg/instance-seg_tensorrt_dynamic-320x320-1344x1344.py
|
deploy_config: configs/mmdet/instance-seg/instance-seg_tensorrt_dynamic-320x320-1344x1344.py
|
||||||
|
|
||||||
pipeline_seg_trt_dynamic_fp16: &pipeline_seg_trt_dynamic_fp16
|
pipeline_seg_trt_dynamic_fp16: &pipeline_seg_trt_dynamic_fp16
|
||||||
|
|
|
@ -429,13 +429,13 @@ def get_fps_metric(shell_res: int, pytorch_metric: dict, metric_info: dict,
|
||||||
metric_key = metric_info[metric_name]['metric_key']
|
metric_key = metric_info[metric_name]['metric_key']
|
||||||
tolerance = metric_info[metric_name]['tolerance']
|
tolerance = metric_info[metric_name]['tolerance']
|
||||||
multi_value = metric_info[metric_name].get('multi_value', 1.0)
|
multi_value = metric_info[metric_name].get('multi_value', 1.0)
|
||||||
compare_flag = True
|
compare_flag = False
|
||||||
output_result[metric_name] = '-'
|
output_result[metric_name] = 'x'
|
||||||
if metric_key in backend_results:
|
if metric_key in backend_results:
|
||||||
backend_value = backend_results[metric_key] * multi_value
|
backend_value = backend_results[metric_key] * multi_value
|
||||||
output_result[metric_name] = backend_value
|
output_result[metric_name] = backend_value
|
||||||
if backend_value < metric_value - tolerance:
|
if backend_value >= metric_value - tolerance:
|
||||||
compare_flag = False
|
compare_flag = True
|
||||||
compare_results[metric_name] = compare_flag
|
compare_results[metric_name] = compare_flag
|
||||||
|
|
||||||
if len(compare_results):
|
if len(compare_results):
|
||||||
|
@ -489,7 +489,7 @@ def get_backend_fps_metric(deploy_cfg_path: str, model_cfg_path: Path,
|
||||||
fps, backend_metric, test_pass = get_fps_metric(return_code,
|
fps, backend_metric, test_pass = get_fps_metric(return_code,
|
||||||
pytorch_metric,
|
pytorch_metric,
|
||||||
metric_info, work_dir)
|
metric_info, work_dir)
|
||||||
logger.info(f'test_pass={test_pass}, results{backend_metric}')
|
logger.info(f'test_pass= {test_pass}, results= {backend_metric}')
|
||||||
metric_list = []
|
metric_list = []
|
||||||
for metric in metric_info:
|
for metric in metric_info:
|
||||||
value = '-'
|
value = '-'
|
||||||
|
|
Loading…
Reference in New Issue