提交 377a7b4b 编写于 作者: W weishengyu

dbg

上级 9a1d35ba
...@@ -67,7 +67,7 @@ class GalleryLayer(paddle.nn.Layer): ...@@ -67,7 +67,7 @@ class GalleryLayer(paddle.nn.Layer):
input_tensor = paddle.zeros(self.image_shape) input_tensor = paddle.zeros(self.image_shape)
gallery_feature = paddle.zeros((len(self.gallery_images), embedding_size)) gallery_feature = paddle.zeros((len(self.gallery_images), embedding_size))
for i, image_path in enumerate(self.gallery_images): 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: for op in preprocess_ops:
image = op(image) image = op(image)
input_tensor[batch_index] = image input_tensor[batch_index] = image
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册