From 37c7612086f1a68b7c5c5e165574066c7f3704fb Mon Sep 17 00:00:00 2001 From: LielinJiang Date: Fri, 24 Apr 2020 12:25:14 +0000 Subject: [PATCH] fix windows label bug --- examples/image_classification/imagenet_dataset.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/image_classification/imagenet_dataset.py b/examples/image_classification/imagenet_dataset.py index 25dcc33..27c41d6 100644 --- a/examples/image_classification/imagenet_dataset.py +++ b/examples/image_classification/imagenet_dataset.py @@ -50,7 +50,7 @@ class ImageNetDataset(DatasetFolder): def __getitem__(self, idx): img_path, label = self.samples[idx] img = cv2.imread(img_path).astype(np.float32) - label = np.array([label]) + label = np.array([label]).astype(np.int64) return self.transform(img), label def __len__(self): -- GitLab