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
8 lines
338 B
8 lines
338 B
from image_embedding_operator_template import ImageEmbeddingOperatorTemplate
|
|
|
|
|
|
def test_image_embedding_operator(model_name='resnet50', img_path='./test_data/test.jpg', dimension=2048):
|
|
op = ResnetImageEmbedding(model_name)
|
|
embedding = op(img_path)
|
|
# get the output shape in README
|
|
assert (dimension,)==embedding[0].shape
|