镜像自地址
https://github.com/xming521/CTAI.git
已同步 2025-12-06 06:36:49 +00:00
45 行
1.5 KiB
Python
45 行
1.5 KiB
Python
import os
|
|
|
|
import SimpleITK as sitk
|
|
import cv2
|
|
import numpy as np
|
|
|
|
from data_set import make
|
|
|
|
|
|
def mkdir(path):
|
|
folder = os.path.exists(path)
|
|
if not folder: # 判断是否存在文件夹如果不存在则创建为文件夹
|
|
os.makedirs(path) # makedirs 创建文件时如果路径不存在会创建这个路径
|
|
|
|
|
|
filename_list = make.get_person_files('../data/all/d2/')
|
|
for i in filename_list:
|
|
pid = i[0]
|
|
print(pid)
|
|
for j in i[1]:
|
|
image = sitk.ReadImage(j)
|
|
image_array = sitk.GetArrayFromImage(image).swapaxes(0, 2)
|
|
image_array = np.rot90(image_array, -1)
|
|
image_array = np.fliplr(image_array).squeeze()
|
|
|
|
# ret, image_array = cv2.threshold(image_array, 150, 255, cv2.THRESH_BINARY)
|
|
mkdir(f'../data/png/{pid}/')
|
|
name = j.replace('.dcm', '').split('/')[-1]
|
|
# cv2.imwrite(f'../data/jpg/{pid}/{name}.jpg', image_array, [int(cv2.IMWRITE_JPEG_QUALITY), 100])
|
|
cv2.imwrite(f'../data/png/{pid}/{name}.png', image_array, (cv2.IMWRITE_PNG_COMPRESSION, 0))
|
|
|
|
# print(filename_list)
|
|
|
|
# for i in filename_list:
|
|
# if '.dcm' in i:
|
|
# image = sitk.ReadImage(data_path + '/' + i)
|
|
# image_array = sitk.GetArrayFromImage(image).swapaxes(0, 2)
|
|
# image_array = np.rot90(image_rray, -1)
|
|
# image_array = np.fliplr(image_array)
|
|
# name = i.replace('.dcm', '')
|
|
# cv2.imwrite(f'{data_path}/{name}_train.png', image_array, (cv2.IMWRITE_PNG_COMPRESSION, 0))
|
|
|
|
# t=cv2.imread('data/out/mask-tttt.png',cv2.IMREAD_GRAYSCALE)
|
|
# print(t)
|