hc/JSCC
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

add dataset.py

Browse Source
This commit is contained in:
chun 2023-12-24 11:42:56 +08:00
parent a5ddb9ced8
commit 9a0103ff51
1 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
dataset.py
View File

@ -14,9 +14,11 @@ def main():
path = './Dataset/ImageNet/{}'.format(phase) path = './Dataset/ImageNet/{}'.format(phase)
print('path is {}'.format(path)) print('path is {}'.format(path))
os.makedirs(path, exist_ok=True) os.makedirs(path, exist_ok=True)
os.system('tar -xvf ./Dataset/ILSVRC2012_img_{}.tar -C {}'.format(phase, path)) print('tar -xf ./Dataset/ILSVRC2012_img_{}.tar -C {}'.format(phase, path))
os.system('tar -xf ./Dataset/ILSVRC2012_img_{}.tar -C {}'.format(phase, path))
for tar in os.listdir(path): for tar in os.listdir(path):
os.system('tar -xvf {}/{} -C {}/{}'.format(path, tar, path, tar)) print('tar -xf {}/{} -C {}/{}'.format(path, tar, path, tar))
os.system('tar -xf {}/{} -C {}/{}'.format(path, tar, path, tar))
os.remove('{}/{}'.format(path, tar)) os.remove('{}/{}'.format(path, tar))