Browse Source
Update py
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
0 additions and
6 deletions
-
image_embedding_resnet50.py
|
|
@ -2,17 +2,11 @@ from towhee import pipe, ops, AutoPipes, AutoConfig |
|
|
|
|
|
|
|
|
@AutoConfig.register |
|
|
@AutoConfig.register |
|
|
class ImageEmbeddingConfig: |
|
|
class ImageEmbeddingConfig: |
|
|
""" |
|
|
|
|
|
For UT |
|
|
|
|
|
""" |
|
|
|
|
|
def __init__(self): |
|
|
def __init__(self): |
|
|
self.model = 'resnet50' |
|
|
self.model = 'resnet50' |
|
|
|
|
|
|
|
|
@AutoPipes.register |
|
|
@AutoPipes.register |
|
|
def ImageEmbedding(config): |
|
|
def ImageEmbedding(config): |
|
|
""" |
|
|
|
|
|
For UT |
|
|
|
|
|
""" |
|
|
|
|
|
return ( |
|
|
return ( |
|
|
pipe.input('path') |
|
|
pipe.input('path') |
|
|
.map('path', 'img', ops.image_decode()) |
|
|
.map('path', 'img', ops.image_decode()) |
|
|
|
