From e677ffc85dc0db33fad70a3b3e0a4c1f09043f2d Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 31 Mar 2020 15:16:29 +0800 Subject: [PATCH] fix(mge/dataset) fix imagenet dir error GitOrigin-RevId: e0a59464bbd69587e0c348ab327762bd1c3440f2 --- python_module/megengine/data/dataset/vision/imagenet.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/python_module/megengine/data/dataset/vision/imagenet.py b/python_module/megengine/data/dataset/vision/imagenet.py index 5e870973..9a5abff7 100644 --- a/python_module/megengine/data/dataset/vision/imagenet.py +++ b/python_module/megengine/data/dataset/vision/imagenet.py @@ -220,5 +220,7 @@ class ImageNet(ImageFolder): logger.info("process devkit file..") untargz(os.path.join(self.root, self.raw_file_meta["devkit"][0])) logger.info("process valid raw file.. this may take 10-20 minutes") - untar(os.path.join(self.root, self.raw_file_meta["val"][0])) + raw_val_dir = os.path.join(self.root, "ILSVRC2012_img_val") + os.makedirs(raw_val_dir, exist_ok=True) + untar(os.path.join(self.root, self.raw_file_meta["val"][0]), raw_val_dir) self.organize_val_data()