diff --git a/src/predict/factory.py b/src/predict/factory.py index f0d854e..74d013d 100644 --- a/src/predict/factory.py +++ b/src/predict/factory.py @@ -15,7 +15,7 @@ def factory_for_gt(im_size, name=None, path_gt=None): with open(path_gt, 'r') as f: dic_names = json.load(f) print('-' * 120 + "\nMonoloco: Ground-truth file opened") - except FileNotFoundError: + except (FileNotFoundError, TypeError): print('-' * 120 + "\nMonoloco: ground-truth file not found\n") dic_names = {}