Browse Source
Update the test script
Signed-off-by: shiyu22 <shiyu.chen@zilliz.com>
main
1 changed files with
0 additions and
2 deletions
-
test_image_embedding_pipeline_yaml.py
|
|
@ -1,6 +1,4 @@ |
|
|
import unittest |
|
|
|
|
|
from towhee import pipeline |
|
|
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): |
|
|
def test_image_embedding_resnet50(repo_name='towhee/image-embedding-pipeline-template', img_path='./test_data/test.jpg', dimension=2048): |
|
|
|
