From 55a5636129b4ec6eb68fb571e00420b50ab10a27 Mon Sep 17 00:00:00 2001 From: Cathy Wong Date: Tue, 14 Jun 2022 11:52:40 -0400 Subject: [PATCH] FaceRecognitionForTracking - fix image transform processing --- research/cv/FaceRecognitionForTracking/eval.py | 2 +- research/cv/FaceRecognitionForTracking/preprocess.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/research/cv/FaceRecognitionForTracking/eval.py b/research/cv/FaceRecognitionForTracking/eval.py index aba67c818..ceee98c3b 100644 --- a/research/cv/FaceRecognitionForTracking/eval.py +++ b/research/cv/FaceRecognitionForTracking/eval.py @@ -104,7 +104,7 @@ def load_images(paths, batch_size=128): im = resize(im) img = np.array(im) ts = transform(img) - ll.append(ts[0]) + ll.append(ts) if len(ll) == batch_size: yield np.stack(ll, axis=0) ll.clear() diff --git a/research/cv/FaceRecognitionForTracking/preprocess.py b/research/cv/FaceRecognitionForTracking/preprocess.py index 140134811..b4c813207 100644 --- a/research/cv/FaceRecognitionForTracking/preprocess.py +++ b/research/cv/FaceRecognitionForTracking/preprocess.py @@ -34,7 +34,7 @@ def load_images(paths, batch_size=1): im = resize(im) img = np.array(im) ts = transform(img) - ll.append(ts[0]) + ll.append(ts) if len(ll) == batch_size: yield np.stack(ll, axis=0) ll.clear() -- Gitee