From 40eddd66e47f1a3730086e6682d8dad6bcb90657 Mon Sep 17 00:00:00 2001 From: charlesbvll Date: Thu, 29 Apr 2021 15:55:55 +0200 Subject: [PATCH] Fixed for test --- monoloco/predict.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/monoloco/predict.py b/monoloco/predict.py index c267fd1..77587ae 100644 --- a/monoloco/predict.py +++ b/monoloco/predict.py @@ -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':