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

update class name

上级 77f46bc1
......@@ -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
......
import importlib
from .det import PPYOLOv2PostPro
from .det import DetPostPro
def build_postprocessor(config):
......
......@@ -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"]
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册