logo

towhee
/
image-embedding-pipeline-template
repo-copy-icon

copied

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
Tree: 3a85c5d920
Branches Tags
${ item.name }
Create tag ${ searchTerm }
Create branch ${ searchTerm }
from '3a85c5d920'
${ noResults }
image-embedding-pipeline-te.../test_image_embedding_pipeli...

10 lines
370 B
Raw Normal View History

Update Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago
import unittest
from towhee import pipeline
from config import DIMENSION, REPO_NAME, TEST_IMG
Update the test script Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
3 years ago
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