Added jp2 image support

pytorch_toolbelt/utils/fs.py

@@ -5,7 +5,7 @@
 import glob
 import os
 from pathlib import Path
-from typing import Union, List
+from typing import Union, List, Iterable
 
 import cv2
 import numpy as np
@@ -26,11 +26,11 @@
     "read_rgb_image",
 ]
 
-COMMON_IMAGE_EXTENSIONS = [".bmp", ".png", ".jpeg", ".jpg", ".tiff", ".tif"]
+COMMON_IMAGE_EXTENSIONS = (".bmp", ".png", ".jpeg", ".jpg", ".tiff", ".tif", ".jp2")
 
 
-def has_ext(fname: str, extensions: Union[str, List[str]]) -> bool:
-    if not isinstance(extensions, (str, list)):
+def has_ext(fname: str, extensions: Union[str, Iterable[str]]) -> bool:
+    if not isinstance(extensions, (str, list, tuple)):
         raise ValueError("Argument extensions must be either string or list of strings")
     if isinstance(extensions, str):
         extensions = [extensions]
@@ -45,6 +45,9 @@ def has_image_ext(fname: str) -> bool:
 
 
 def find_in_dir(dirname: str) -> List[str]:
+    if not os.path.isdir(dirname):
+        raise FileNotFoundError(f"Directory {dirname} not found")
+
     return [os.path.join(dirname, fname) for fname in sorted(os.listdir(dirname))]