Browse Source
Update the test script
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
3 additions and
2 deletions
-
test_image_embedding_pipeline_yaml.py
|
|
|
@ -3,5 +3,6 @@ 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) |
|
|
|
assert (1, dimension)==embedding[0].shape |
|
|
|
(embedding,) = embedding_pipeline(img_path) |
|
|
|
# get the output shape in README |
|
|
|
assert (dimension,)==embedding[0].shape |
|
|
|
|
