convert image to RGB image.

3 years ago · 14cacc0ee7
1 changed files with 1 additions and 1 deletions
--- 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)