提交 693a962c 编写于 作者: W weishengyu

update class name

上级 77f46bc1
...@@ -23,7 +23,7 @@ Modules: ...@@ -23,7 +23,7 @@ Modules:
inference_model_dir: ./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer/ inference_model_dir: ./models/ppyolov2_r50vd_dcn_mainbody_v1.0_infer/
input_names: input_names:
output_names: output_names:
- name: PPYOLOv2PostPro - name: DetPostPro
type: postprocessor type: postprocessor
threshold: 0.2 threshold: 0.2
max_det_results: 1 max_det_results: 1
......
import importlib import importlib
from .det import PPYOLOv2PostPro from .det import DetPostPro
def build_postprocessor(config): def build_postprocessor(config):
......
...@@ -5,7 +5,7 @@ from utils import logger ...@@ -5,7 +5,7 @@ from utils import logger
from ...base_processor import BaseProcessor from ...base_processor import BaseProcessor
class PPYOLOv2PostPro(BaseProcessor): class DetPostPro(BaseProcessor):
def __init__(self, config): def __init__(self, config):
self.threshold = config["threshold"] self.threshold = config["threshold"]
self.label_list = config["label_list"] self.label_list = config["label_list"]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册