From 166d74309f060d23d002dc60308bbf870d0ab753 Mon Sep 17 00:00:00 2001 From: Kensuke Matsuzaki Date: Mon, 13 Feb 2023 12:18:12 +0900 Subject: [PATCH] Specify dtype of verification data --- recognition/arcface_torch/eval/verification.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/recognition/arcface_torch/eval/verification.py b/recognition/arcface_torch/eval/verification.py index edacf8d81..4d355a1a7 100644 --- a/recognition/arcface_torch/eval/verification.py +++ b/recognition/arcface_torch/eval/verification.py @@ -206,7 +206,7 @@ def load_bin(path, image_size): bins, issame_list = pickle.load(f, encoding='bytes') # py3 data_list = [] for flip in [0, 1]: - data = torch.empty((len(issame_list) * 2, 3, image_size[0], image_size[1])) + data = torch.zeros((len(issame_list) * 2, 3, image_size[0], image_size[1]), dtype=torch.uint8) data_list.append(data) for idx in range(len(issame_list) * 2): _bin = bins[idx]