chore: support .dat rgb file extension (#155)

TRI-ML · Jun 23, 2023 · 4608c9e · 4608c9e
1 parent bfd93e8
commit 4608c9e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/dgp/utils/dataset_conversion.py b/dgp/utils/dataset_conversion.py
@@ -83,6 +83,7 @@ def compute_image_statistics(image_list, image_open_fn, single_process=False, nu
         ".bmp",
         ".pgm",
         ".tiff",
+        ".dat",
     )
     image_list = list(filter(lambda x: x.lower().endswith(valid_extensions), image_list))
     if not image_list: