Fix for list inputs

Signed-off-by: Jael Gu <mengjia.gu@zilliz.com>
1 year ago · 3a82fb4ebe
1 changed files with 1 additions and 1 deletions
--- a/timm_image.py
+++ b/timm_image.py
@ -125,7 +125,7 @@ class TimmImage(NNOperator):
            vecs = [list(x.detach().numpy()) for x in features] if isinstance(features, list) \
                else list(features.detach().numpy())
        else:
-            vecs = [x.squeeze(0).detach().numpy()] if isinstance(features, list) \
+            vecs = [x.squeeze(0).detach().numpy() for x in features] if isinstance(features, list) \
                else features.squeeze(0).detach().numpy()
        return vecs