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
9 lines
370 B
9 lines
370 B
import unittest
|
|
from towhee import pipeline
|
|
from config import DIMENSION, REPO_NAME, TEST_IMG
|
|
|
|
|
|
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
|