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

update some names

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