charles/monoloco
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixed for test

Browse Source
This commit is contained in:
charlesbvll 2021-04-29 15:55:55 +02:00
parent f52703b795
commit 40eddd66e4
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
monoloco/predict.py
View File

@ -251,7 +251,7 @@ def factory_outputs(args, pifpaf_outs, dic_out, output_path, kk=None):
else:
assert 'json' in args.output_types or args.mode == 'keypoints', \
"No output saved, please select one among front, bird, multi, json, or pifpaf arguments"
if args.social_distance:
if args.activities and 'social_distance' in args.activities:
assert args.mode == 'mono', "Social distancing only works with monocular network"
if args.mode == 'keypoints':