diff --git a/deploy/pphuman/datacollector.py b/deploy/pphuman/datacollector.py index f1e3a21360fb871e26e53129cd8833cd123f1422..6bf4fee8531165a97b59112057c2abb8fb893389 100644 --- a/deploy/pphuman/datacollector.py +++ b/deploy/pphuman/datacollector.py @@ -90,7 +90,7 @@ class DataCollector(object): if kpt_res: self.collector[ids]["kpts"].append( [kpt_res['keypoint'][0][idx], kpt_res['keypoint'][1][idx]]) - if action_res: + if action_res and (idx + 1) in action_res: self.collector[ids]["actions"].append(action_res[idx + 1]) else: # action model generate result per X frames, Not available every frames diff --git a/deploy/pphuman/pipe_utils.py b/deploy/pphuman/pipe_utils.py index b55ac9a867cf027662d9b3d84e39f119f28d123a..43904e657c1a4921adb6656ecb25682aa7d0f984 100644 --- a/deploy/pphuman/pipe_utils.py +++ b/deploy/pphuman/pipe_utils.py @@ -217,7 +217,7 @@ class PipeTimer(Times): def merge_model_dir(args, model_dir): # set --model_dir DET=ppyoloe/ to overwrite the model_dir in config file - task_set = ['DET', 'ATTR', 'MOT', 'KPT', 'ACTION'] + task_set = ['DET', 'ATTR', 'MOT', 'KPT', 'ACTION', 'REID'] if not model_dir: return args for md in model_dir: diff --git a/deploy/pphuman/reid.py b/deploy/pphuman/reid.py index cef4029239f7e0f635547506282c2527bf687353..3f4d59d78e8273385353a45248a059d523eb478c 100644 --- a/deploy/pphuman/reid.py +++ b/deploy/pphuman/reid.py @@ -73,7 +73,7 @@ class ReID(object): self.det_times = Timer() self.cpu_mem, self.gpu_mem, self.gpu_util = 0, 0, 0 self.batch_size = batch_size - self.input_wh = [128, 256] + self.input_wh = (128, 256) def set_config(self, model_dir): return PredictConfig(model_dir)