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 13:17:52 +08:00
parent 5ab9422ce6
commit 579d2b0d8e
1 changed files with 6 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
dataset.py
View File

@ -16,11 +16,12 @@ def main():
os.makedirs(path, exist_ok=True)
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):
print('tar -xf {}/{} -C {}/{}'.format(path, tar, path, tar.split('.')[0]))
os.makedirs('{}/{}'.format(path, tar.split('.')[0]), exist_ok=True)
os.system('tar -xf {}/{} -C {}/{}'.format(path, tar, path, tar.split('.')[0]))
os.remove('{}/{}'.format(path, tar))
if phase == 'train':
for tar in os.listdir(path):
print('tar -xf {}/{} -C {}/{}'.format(path, tar, path, tar.split('.')[0]))
os.makedirs('{}/{}'.format(path, tar.split('.')[0]), exist_ok=True)
os.system('tar -xf {}/{} -C {}/{}'.format(path, tar, path, tar.split('.')[0]))
os.remove('{}/{}'.format(path, tar))
if __name__ == '__main__':