logo

towhee
/
resnet-image-embedding
repo-copy-icon

copied

Browse Source

convert image to RGB image.

training
wxywb 3 years ago
parent
be15aa6e96
commit
14cacc0ee7
1 changed files with 1 additions and 1 deletions
Whitespace
Show all changes Ignore whitespace when comparing lines Ignore changes in amount of whitespace Ignore changes in whitespace at EOL
Unified View
Diff Options
Show Stats Download Patch File Download Diff File
  1. 2
      resnet_image_embedding.py

2
resnet_image_embedding.py
View File

@ -45,7 +45,7 @@ class ResnetImageEmbedding(Operator):
def __call__(self, img_path: str) -> NamedTuple('Outputs', [('feature_vector', numpy.ndarray)]): 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) embedding = self.model(img)
Outputs = NamedTuple('Outputs', [('feature_vector', numpy.ndarray)]) Outputs = NamedTuple('Outputs', [('feature_vector', numpy.ndarray)])
return Outputs(embedding) return Outputs(embedding)

Write Preview
Loading…
Cancel
Save