Browse Source

fix(mge/dataset) fix imagenet dir error

GitOrigin-RevId: e0a59464bb
tags/v0.3.2
Megvii Engine Team 5 years ago
parent
commit
e677ffc85d
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      python_module/megengine/data/dataset/vision/imagenet.py

+ 3
- 1
python_module/megengine/data/dataset/vision/imagenet.py View File

@@ -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()

Loading…
Cancel
Save