From c9d3dcd1fbad0774211d708e6a0754847261e70c Mon Sep 17 00:00:00 2001 From: shiyu22 Date: Thu, 9 Dec 2021 17:56:08 +0800 Subject: [PATCH] Update Signed-off-by: shiyu22 --- resnet50_image_embedding.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/resnet50_image_embedding.py b/resnet50_image_embedding.py index 66f3bdb..8e9a7cf 100644 --- a/resnet50_image_embedding.py +++ b/resnet50_image_embedding.py @@ -31,8 +31,6 @@ class Resnet50ImageEmbedding(Operator): sys.path.append(str(Path(__file__).parent)) if framework == 'pytorch': from pytorch.model import Model - if framework == 'tensorflow': - from tensorflow.model import Model self.model = Model(model_name) def __call__(self, img_tensor: torch.Tensor) -> NamedTuple('Outputs', [('cnn', numpy.ndarray)]):