Spaces:

weizhongyuan123
/

ALIKE

Runtime error

Zhao Xiaoming commited on Jun 8, 2022

Commit

688064e

1 Parent(s): 35bfce9

BGR2RGB

Files changed (1) hide show

demo.py CHANGED Viewed

@@ -55,7 +55,7 @@ class ImageLoader(object):
             img = cv2.imread(filename)
             if img is None:
                 raise Exception('Error reading image %s' % filename)
         return img
     def __len__(self):

             img = cv2.imread(filename)
             if img is None:
                 raise Exception('Error reading image %s' % filename)
+        img = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
         return img
     def __len__(self):