From 8f2390fc14c989ead30e4f9b74a49e8c668b12db Mon Sep 17 00:00:00 2001 From: tpoisonooo Date: Sat, 15 Oct 2022 11:41:09 +0800 Subject: [PATCH] fix(imperative): untar imagenet use bad path --- imperative/python/megengine/data/dataset/vision/imagenet.py | 6 ++---- 1 file changed, 2 insertions(+), 4 deletions(-) diff --git a/imperative/python/megengine/data/dataset/vision/imagenet.py b/imperative/python/megengine/data/dataset/vision/imagenet.py index ad775eaa..b36d213a 100644 --- a/imperative/python/megengine/data/dataset/vision/imagenet.py +++ b/imperative/python/megengine/data/dataset/vision/imagenet.py @@ -205,7 +205,7 @@ class ImageNet(ImageFolder): calculate_md5(raw_file) == checksum ), "checksum mismatch, {} may be damaged".format(raw_file) logger.info("extract valid tar file... this may take 10-20 minutes") - untar(os.path.join(self.root, raw_file), self.target_folder) + untar(raw_file, self.target_folder) self._organize_val_data() def _prepare_train(self): @@ -217,9 +217,7 @@ class ImageNet(ImageFolder): calculate_md5(raw_file) == checksum ), "checksum mismatch, {} may be damaged".format(raw_file) logger.info("extract train tar file.. this may take several hours") - untar( - os.path.join(self.root, raw_file), self.target_folder, - ) + untar(raw_file, self.target_folder) paths = [ os.path.join(self.target_folder, child_dir) for child_dir in os.listdir(self.target_folder)