From 6f338339672b64b1f171620d60a02e5097be4513 Mon Sep 17 00:00:00 2001 From: weishengyu Date: Tue, 25 Jan 2022 17:40:48 +0800 Subject: [PATCH] dbg --- ppcls/utils/gallery2fc.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ppcls/utils/gallery2fc.py b/ppcls/utils/gallery2fc.py index ca2084082..67b08529e 100644 --- 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