给图片添加分割线

我的心永远是冰冰哒2024-06-08 16:22

复制代码

import pandas as pd
import SimpleITK as sitk
import os
import pydicom
import numpy as np 
import matplotlib.pyplot as plt 
import nrrd 
import matplotlib.image as mpimg
from PIL import Image 
import glob
import cv2 
import pickle 

jpg_seg_read_path = "/Users/yxk/Desktop/核磁项目/ALL_sample/seg_pic/安芮716608_000007_seg.jpg"
seg_image = Image.open(jpg_seg_read_path)
seg_img_arr = np.array(seg_image)
print(seg_img_arr.shape)

print(np.sum(seg_img_arr[:,:,0] != seg_img_arr[:,:,2]))
print(np.sum(seg_img_arr[:,:,0] != seg_img_arr[:,:,1]))
print(np.sum(seg_img_arr[:,:,1] != seg_img_arr[:,:,2]))

plt.imshow(seg_img_arr,cmap='gray')
# seg_img_arr

给三维的图片添加分割线

复制代码

seg_img_arr[100:300,200,:]=[255,0,0] # 垂直线：x=200
seg_img_arr[100:300,300,:]=[0,255,0] # 垂直线：x=300
seg_img_arr[100,200:300,:]=[0,0,255] # 垂直线：x=300
seg_img_arr[300,200:300,:]=[0,0,255] # 垂直线：x=300
plt.imshow(seg_img_arr,cmap='gray')
    
# img_arr[mask_pos[:,:,0],:] = [255,0,0] ## 注意numpy array是可以一次性赋值的，不需要一个个赋值
# pil_img = Image.fromarray(img_arr)
# #dcm_id,_ = file.split(".")
# pil_img.save(save_mask_dir+ patient + str(id) + "_" +'{0:06d}'.format(i)+ "_mask.jpg")