![](https://img-blog.csdnimg.cn/9e0bace0c4ca49c78e4935bfc35376e7.png?x-oss-process=image/resize,m_fixed,h_224,w_224)
图像分割
tensorflow2
吴天德少侠
这个作者很懒,什么都没留下…
展开
-
tensorflow2的utils.Sequence
假设文件是这样的images和labels里面保存的都是.npy数组images里面的一个数据的shape=[128,128,16,1],labels里面的一个数据的shape=[128,128,16,2],因为是二分类语义分割data_loader.pyfrom tensorflow.keras.utils import Sequenceimport numpy as npimport math# Here, `x_set` is list of path to the原创 2022-03-15 23:19:39 · 964 阅读 · 0 评论 -
将numpy数组保存成png标签
# 将图像一张一张抽出来保存成png标签import numpy as npimport SimpleITK as sitkimport matplotlib.pyplot as pltimport osfrom PIL import Imagefrom tqdm import tqdmif __name__ == '__main__': image_path = "D:/test/CASE01_Segmentation.nii" filename = os.path..原创 2022-03-11 16:44:46 · 6886 阅读 · 0 评论 -
将图像分割结果以彩色图展示
由于实际项目中没有那么多类,所以就假设有10个种类,每个种类设置对应一种颜色import numpy as npfrom PIL import Imageimport matplotlib.pyplot as pltif __name__ == '__main__': # input_shape = [5,5] img = np.random.randint(0,10,size=input_shape,dtype='uint8') plt.subplot(原创 2022-03-02 17:40:46 · 3259 阅读 · 0 评论