|
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() |