From be15aa6e9608062e2731164399d028e5ddc20717 Mon Sep 17 00:00:00 2001 From: Jael Gu Date: Sat, 25 Dec 2021 15:04:46 +0800 Subject: [PATCH 1/2] Update requirements.txt Signed-off-by: Jael Gu --- requirements.txt | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/requirements.txt b/requirements.txt index e69de29..eda3109 100644 --- a/requirements.txt +++ b/requirements.txt @@ -0,0 +1,4 @@ +torch==1.9.0 +torchvision==0.10.0 +pillow==8.3.1 +numpy==1.19.5 From 14cacc0ee7574ab6843088562e64d176d58dcb99 Mon Sep 17 00:00:00 2001 From: wxywb Date: Mon, 27 Dec 2021 11:40:21 +0800 Subject: [PATCH 2/2] convert image to RGB image. --- resnet_image_embedding.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/resnet_image_embedding.py b/resnet_image_embedding.py index ba18e2d..f11d958 100644 --- a/resnet_image_embedding.py +++ b/resnet_image_embedding.py @@ -45,7 +45,7 @@ class ResnetImageEmbedding(Operator): def __call__(self, img_path: str) -> NamedTuple('Outputs', [('feature_vector', numpy.ndarray)]): - img = self.tfms(Image.open(img_path)).unsqueeze(0) + img = self.tfms(Image.open(img_path).convert('RGB')).unsqueeze(0) embedding = self.model(img) Outputs = NamedTuple('Outputs', [('feature_vector', numpy.ndarray)]) return Outputs(embedding)