logo
Browse Source

Add image_embedding_resnet50.py

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>

Add image_embedding_resnet50.py

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>

Add image_embedding_resnet50.py

Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
shiyu22 2 years ago
parent
commit
1186063eed
  1. 21
      image_embedding_resnet50.py

21
image_embedding_resnet50.py

@ -0,0 +1,21 @@
from towhee import pipe, ops, AutoPipes, AutoConfig
@AutoConfig.register
class ImageEmbeddingConfig:
"""
For UT
"""
def __init__(self):
self.model = 'resnet50'
@AutoPipes.register
def ImageEmbedding(config):
"""
For UT
"""
return (
pipe.input('path')
.map('path', 'img', ops.image_decode())
.map('img', 'embedding', ops.image_embedding.timm(model_name=config.model))
.output('embedding')
)
Loading…
Cancel
Save