Condition keypoint validity check on input param (albumentations-team…

…#216)
linghai06 · Mar 4, 2019 · 613ef68 · 613ef68
1 parent 3230437
commit 613ef68
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/albumentations/core/composition.py b/albumentations/core/composition.py
@@ -198,7 +198,8 @@ def data_postprocessing(data_name, params, check_fn, filter_fn, convert_fn, data
     if params['format'] == 'albumentations':
         check_fn(data[data_name])
     else:
-        data[data_name] = convert_fn(data[data_name], params['format'], rows, cols, check_validity=True)
+        data[data_name] = convert_fn(data[data_name], params['format'], rows, cols,
+                                     check_validity=bool(params.get('remove_invisible', True)))
 
     data = remove_label_fields_from_data(data_name, params.get('label_fields', []), data)
     return data