Browse Source
Update
Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
test
1 changed files with
1 additions and
2 deletions
-
timm_image.py
|
@ -30,10 +30,9 @@ from timm.models.factory import create_model |
|
|
import warnings |
|
|
import warnings |
|
|
|
|
|
|
|
|
warnings.filterwarnings('ignore') |
|
|
warnings.filterwarnings('ignore') |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
log = logging.getLogger() |
|
|
log = logging.getLogger() |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@register(output_schema=['vec']) |
|
|
@register(output_schema=['vec']) |
|
|
class TimmImage(NNOperator): |
|
|
class TimmImage(NNOperator): |
|
|
""" |
|
|
""" |
|
|