From 693a962c2df87d2abda4f7ee965b8347dbcd96f0 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Fri, 11 Mar 2022 20:29:20 +0800 Subject: [PATCH] update class name --- deploy/python/ppshitu_v2/configs/test_config.yml | 2 +- .../ppshitu_v2/processor/algo_mod/postprocessor/__init__.py | 2 +- .../python/ppshitu_v2/processor/algo_mod/postprocessor/det.py | 2 +- 3 files changed, 3 insertions(+), 3 deletions(-) diff --git a/deploy/python/ppshitu_v2/configs/test_config.yml b/deploy/python/ppshitu_v2/configs/test_config.yml index 6358934c..e230bb1d 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 b7284783..89500a7d 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 1c6d410b..c69a9cf1 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"] -- GitLab