提交 653009ab 编写于 作者: Z zhiboniu 提交者: zhiboniu

update some names

上级 bb8a2b02
......@@ -9,14 +9,13 @@ DET:
batch_size: 1
MOT:
model_dir: output_inference/mot_ppyoloe_s_36e_ppvehicle/
model_dir: output_inference/mot_ppyoloe_l_36e_ppvehicle/
tracker_config: deploy/pphuman/config/tracker_config.yml
batch_size: 1
basemode: "idbased"
enable: False
VEHICLE_PLATE:
det_algorithm: "DB"
det_model_dir: output_inference/ch_PP-OCRv3_det_infer/
det_limit_side_len: 480
det_limit_type: "max"
......
......@@ -82,7 +82,7 @@ class DataCollector(object):
kpt_res = Result.get('kpt')
skeleton_action_res = Result.get('skeleton_action')
reid_res = Result.get('reid')
vehicplate_res = Result.get('vehicleplate')
vehicleplate_res = Result.get('vehicleplate')
rects = []
if reid_res is not None:
......@@ -112,8 +112,8 @@ class DataCollector(object):
idx])
self.collector[ids]["qualities"].append(reid_res['qualities'][
idx])
if vehicplate_res:
self.collector[ids]["vehicleplate"].append(vehicplate_res[
if vehicleplate_res:
self.collector[ids]["vehicleplate"].append(vehicleplate_res[
'plate'][idx])
def get_res(self):
......
......@@ -37,7 +37,6 @@ from pphuman.pipe_utils import merge_cfg, print_arguments, argsparser
class PlateDetector(object):
def __init__(self, args, cfg):
self.args = args
self.det_algorithm = cfg['det_algorithm']
self.pre_process_list = {
'Resize_Mult32': {
'limit_side_len': cfg['det_limit_side_len'],
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册