modified - TestDataset.py; include .exr in img_fmts

modified - imutils.py; add opencv exr enable envirionment to consider .exr
; add imread for .exr image
; modified condition from 16-bit check to non 8-bit check

lib/common/imutils.py

@@ -1,3 +1,5 @@
+import os
+os.environ["OPENCV_IO_ENABLE_OPENEXR"]="1"
 import cv2
 import mediapipe as mp
 import torch
@@ -51,10 +53,13 @@ def get_affine_matrix_box(boxes, w2, h2):
 
 def load_img(img_file):
 
-    img = cv2.imread(img_file, cv2.IMREAD_UNCHANGED)
+    if img_file.endswith("exr"):
+        img = cv2.imread(img_file, cv2.IMREAD_ANYCOLOR | cv2.IMREAD_ANYDEPTH)  
+    else :
+        img = cv2.imread(img_file, cv2.IMREAD_UNCHANGED)
 
-    # considering 16-bit image
-    if img.dtype == np.uint16:
+    # considering non 8-bit image
+    if img.dtype != np.uint8 :
         img = cv2.normalize(img, None, 0, 255, cv2.NORM_MINMAX, dtype=cv2.CV_8U)
 
     if len(img.shape) == 2:

lib/dataset/TestDataset.py

@@ -64,7 +64,7 @@ class TestDataset:
         self.device = device
 
         keep_lst = sorted(glob.glob(f"{self.image_dir}/*"))
-        img_fmts = ["jpg", "png", "jpeg", "JPG", "bmp"]
+        img_fmts = ["jpg", "png", "jpeg", "JPG", "bmp", "exr"]
 
         self.subject_list = sorted(
             [item for item in keep_lst if item.split(".")[-1] in img_fmts], reverse=False