Limit masks to supported extensions

opendm/context.py

@@ -51,7 +51,7 @@ odm_modules_src_path = os.path.join(root_path, "modules")
 settings_path = os.path.join(root_path, 'settings.yaml')
 
 # Define supported image extensions
-supported_extensions = {'.jpg','.jpeg','.png', '.tif', '.tiff'}
+supported_extensions = {'.jpg','.jpeg','.png', '.tif', '.tiff', '.bmp'}
 
 # Define the number of cores
 num_cores = multiprocessing.cpu_count()

stages/dataset.py

@@ -99,7 +99,7 @@ class ODMLoadDatasetStage(types.ODM_Stage):
                 masks = {}
                 for r in rejects:
                     (p, ext) = os.path.splitext(r)
-                    if p[-5:] == "_mask":
+                    if p[-5:] == "_mask" and ext.lower() in context.supported_extensions:
                         masks[p] = r
 
                 photos = []