dbg

2022-01-25 17:40:48 +08:00 · 2022-01-25 17:40:48 +08:00 · 6f33833967
parent e95dcec819
commit 6f33833967
1 changed files with 1 additions and 1 deletions
--- a/ppcls/utils/gallery2fc.py
+++ b/ppcls/utils/gallery2fc.py
@ -73,7 +73,7 @@ class GalleryLayer(paddle.nn.Layer):
        input_tensor = paddle.zeros(self.image_shape)
        gallery_feature = paddle.zeros((len(self.gallery_images), embedding_size))
        for i, image_path in enumerate(self.gallery_images):
-            image = cv2.imread(image_path)
+            image = cv2.imread(image_path)[:, :, ::-1]
            for op in preprocess_ops:
                image = op(image)
            input_tensor[batch_index] = image