[b52eda]: / Pack_Dataset.py

Download this file

19 lines (15 with data), 595 Bytes

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