image-embedding-pipeline-template/test_image_embedding_pipeline_yaml.py

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