Loading colors correctly

#3 finished
radxtools · Dec 6, 2020 · db3ae8d · db3ae8d
1 parent 980c768
commit db3ae8d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/utils/primitives.py b/utils/primitives.py
@@ -247,6 +247,9 @@ def copy_attributes(self, other: 'Image'):
     @staticmethod
     def load(path: Union[str, Path]):
         img = cv2.imread(str(path))
+        bands = len(img.shape)
+        if(bands == 3):
+            img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         return Image(img)
 
     def __repr__(self):