Diff of /Pack_Dataset.py [000000] .. [b52eda]

Switch to unified view

a b/Pack_Dataset.py
1
import os
2
import tarfile as tar
3
from tqdm import tqdm
4
5
PLANES = os.listdir("D:\\ImageCHD_dataset\\")
6
PLANES.remove("ImageCHD_dataset")
7
print(PLANES)
8
9
archive = tar.open("D:\\ImageCHD_dataset.tar.gz", 'x:gz')
10
11
for plane in PLANES:
12
    dirs = os.listdir("D:\\ImageCHD_dataset\\" + plane + "\\")
13
    for dir in dirs:
14
        temp = os.listdir("D:\\ImageCHD_dataset\\" + plane + "\\" + dir + "\\")
15
        print("D:\\ImageCHD_dataset\\" + plane + "\\" + dir + "\\")
16
        for file in tqdm(temp):
17
            archive.add("D:\\ImageCHD_dataset\\" + plane + "\\" + dir + "\\" + file)
18
19
archive.close()