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:40:47 +08:00
parent 0818c20233
commit a5ddb9ced8
1 changed files with 4 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
dataset.py
View File

@ -10,12 +10,15 @@ def main():
raise Exception('not find dataset') raise Exception('not find dataset')
phases = ['train', 'val'] phases = ['train', 'val']
for phase in phases: for phase in phases:
print("extracting {} dataset".format(phase))
path = './Dataset/ImageNet/{}'.format(phase) path = './Dataset/ImageNet/{}'.format(phase)
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)) os.system('tar -xvf ./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)) os.system('tar -xvf {}/{} -C {}/{}'.format(path, tar, path, tar))
os.remove('{}/{}'.format(path, tar)) os.remove('{}/{}'.format(path, tar))
if __name__ == '__main__': if __name__ == '__main__':
main() main()