提交 6f338339 编写于 作者: W weishengyu

dbg

上级 e95dcec8
......@@ -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
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册