towhee
/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
Readme
Files and versions
8 lines
342 B
8 lines
342 B
from towhee import pipeline
|
|
|
|
|
|
def test_image_embedding_resnet50(repo_name='towhee/image-embedding-pipeline-template', img_path='./test_data/test.jpg', dimension=2048):
|
|
embedding_pipeline = pipeline(repo_name)
|
|
(embedding,) = embedding_pipeline(img_path)
|
|
# get the output shape in README
|
|
assert (dimension,)==embedding[0].shape
|