提交 e677ffc8 编写于 作者: M Megvii Engine Team

fix(mge/dataset) fix imagenet dir error

GitOrigin-RevId: e0a59464bbd69587e0c348ab327762bd1c3440f2
上级 c9830d20
......@@ -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()
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册