diff --git a/deploy/python/ppshitu_v2/configs/test_config.yml b/deploy/python/ppshitu_v2/configs/test_config.yml index 6358934c7c48e3f1b528274a5b67ecf44158af6e..e230bb1deb8d067fda8c17d9a9a2142ba9eed050 100644 --- a/deploy/python/ppshitu_v2/configs/test_config.yml +++ b/deploy/python/ppshitu_v2/configs/test_config.yml @@ -23,7 +23,7 @@ Modules: inference_model_dir: ./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer/ input_names: output_names: - - name: PPYOLOv2PostPro + - name: DetPostPro type: postprocessor threshold: 0.2 max_det_results: 1 diff --git a/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/__init__.py b/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/__init__.py index b728478312e78b68459f19d177b77a4bbe280133..89500a7d4b8668ea0c849f16e18c95f69108bf0e 100644 --- a/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/__init__.py +++ b/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/__init__.py @@ -1,6 +1,6 @@ import importlib -from .det import PPYOLOv2PostPro +from .det import DetPostPro def build_postprocessor(config): diff --git a/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/det.py b/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/det.py index 1c6d410be1e30b6b6695d3358895195dd86144ab..c69a9cf1762b7c526b3021ee877801e7aab7eb56 100644 --- a/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/det.py +++ b/deploy/python/ppshitu_v2/processor/algo_mod/postprocessor/det.py @@ -5,7 +5,7 @@ from utils import logger from ...base_processor import BaseProcessor -class PPYOLOv2PostPro(BaseProcessor): +class DetPostPro(BaseProcessor): def __init__(self, config): self.threshold = config["threshold"] self.label_list = config["label_list"]