import os import numpy as np import nibabel as nib import imageio import cv2 def read_niifile(niifilepath): # 读取niifile文件 img = nib.load(niifilepath) # 提取niifile文件 img_fdata = img.get_fdata(dtype='float32') return img_fdata def save_fig(niifilepath, savepath, num, name): # 保存为图片 name = name.split('-')[1] filepath_seg = niifilepath + "segmentation\" + "segmentation-" + name filepath_vol = niifilepath + "volume\" + "volume-" + name savepath_seg = savepath + "segmentation\" savepath_vol = savepath + "volume\" if not os.path.exists(savepath_seg): os.makedirs(savepath_seg) if not os.path.exists(savepath_vol): os.makedirs(savepath_vol) fdata_vol = read_niifile(filepath_vol) fdata_seg = read_niifile(filepath_seg) (x, y, z) = fdata_seg.shape total = x * y for k in range(z): silce_seg = fdata_seg[:, :, k] # 三个位置表示三个不同角度的切片 if silce_seg.max() == 0: continue else: silce_seg = (silce_seg - silce_seg.min()) / (silce_seg.max() - silce_seg.min()) * 255 silce_seg = cv2.threshold(silce_seg, 1, 255, cv2.THRESH_BINARY)[1] if (np.sum(silce_seg == 255) / total) > 0.015: silce_vol = fdata_vol[:, :, k] silce_vol = (silce_vol - silce_vol.min()) / (silce_vol.max() - silce_vol.min()) * 255 imageio.imwrite(os.path.join(savepath_seg, '{}.png'.format(num)), silce_seg) imageio.imwrite(os.path.join(savepath_vol, '{}.png'.format(num)), silce_vol) num += 1 # 将切片信息保存为png格式 return num if name == 'main': path = r"C:\Users\Administrator\Desktop\LiTS2017" savepath = r"C:\Users\Administrator\Desktop\2D-LiTS2017" filenames = os.listdir(path + "segmentation") num = 0 for filename in filenames: num = save_fig(path, savepath, num, filename) 替换掉代码中的cv2模块,实现相同功能
时间: 2023-12-11 18:04:32 浏览: 41
可以使用PIL库替换掉cv2模块,实现相同的功能。具体替换方法如下:
1.导入PIL库
```python
from PIL import Image
```
2.将cv2.threshold替换为PIL.ImageOps.autocontrast
```python
silce_seg = (silce_seg - silce_seg.min()) / (silce_seg.max() - silce_seg.min()) * 255
silce_seg = Image.fromarray(silce_seg.astype('uint8'))
silce_seg = ImageOps.autocontrast(silce_seg, cutoff=1)
silce_seg = np.array(silce_seg)
```
3.将cv2.imwrite替换为PIL.Image.save
```python
imageio.imwrite(os.path.join(savepath_seg, '{}.png'.format(num)), silce_seg)
imageio.imwrite(os.path.join(savepath_vol, '{}.png'.format(num)), silce_vol)
```
替换后的代码如下:
```python
import os
import numpy as np
import nibabel as nib
import imageio
from PIL import Image, ImageOps
def read_niifile(niifilepath):
# 读取niifile文件
img = nib.load(niifilepath)
# 提取niifile文件
img_fdata = img.get_fdata(dtype='float32')
return img_fdata
def save_fig(niifilepath, savepath, num, name):
# 保存为图片
name = name.split('-')[1]
filepath_seg = niifilepath + "segmentation/" + "segmentation-" + name
filepath_vol = niifilepath + "volume/" + "volume-" + name
savepath_seg = savepath + "segmentation/"
savepath_vol = savepath + "volume/"
if not os.path.exists(savepath_seg):
os.makedirs(savepath_seg)
if not os.path.exists(savepath_vol):
os.makedirs(savepath_vol)
fdata_vol = read_niifile(filepath_vol)
fdata_seg = read_niifile(filepath_seg)
(x, y, z) = fdata_seg.shape
total = x * y
for k in range(z):
silce_seg = fdata_seg[:, :, k] # 三个位置表示三个不同角度的切片
if silce_seg.max() == 0:
continue
else:
silce_seg = (silce_seg - silce_seg.min()) / (silce_seg.max() - silce_seg.min()) * 255
silce_seg = Image.fromarray(silce_seg.astype('uint8'))
silce_seg = ImageOps.autocontrast(silce_seg, cutoff=1)
silce_seg = np.array(silce_seg)
if (np.sum(silce_seg == 255) / total) > 0.015:
silce_vol = fdata_vol[:, :, k]
silce_vol = (silce_vol - silce_vol.min()) / (silce_vol.max() - silce_vol.min()) * 255
imageio.imwrite(os.path.join(savepath_seg, '{}.png'.format(num)), silce_seg)
imageio.imwrite(os.path.join(savepath_vol, '{}.png'.format(num)), silce_vol)
num += 1
# 将切片信息保存为png格式
return num
if __name__ == '__main__':
path = r"C:\Users\Administrator\Desktop\LiTS2017"
savepath = r"C:\Users\Administrator\Desktop\2D-LiTS2017"
filenames = os.listdir(path + "segmentation")
num = 0
for filename in filenames:
num = save_fig(path, savepath, num, filename)
```
相关推荐
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![py](https://img-home.csdnimg.cn/images/20210720083646.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![](https://csdnimg.cn/download_wenku/file_type_ask_c1.png)
![pdf](https://img-home.csdnimg.cn/images/20210720083512.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![rar](https://img-home.csdnimg.cn/images/20210720083606.png)
![zip](https://img-home.csdnimg.cn/images/20210720083736.png)
![docx](https://img-home.csdnimg.cn/images/20210720083331.png)