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

5
dataset.py
View File

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