利用CSV路径文件和.png图像,生成3D原图。并展示部分分割图像

具体代码 ,请看的的github

if __name__ == "__main__":
  df = pd.read_csv(r'D:/compation/kaggle/train.csv/train.csv')  # 路径文件
  list_images = glob(r'D:\\compation\\kaggle\\train\\*\\*\\scans\\*.png')  # 利用glob列出所有.png文件
  # print(df.head())
  mask_data_not_null = df[df['segmentation'].notnull()]  # 只提取有分割结果的mask
  print(mask_data_not_null)
  index_list = list(mask_data_not_null.index)
  image_details = show_image(list_images,show=True,create_nii=False)   # 展示image,可以选择是否保存成nii文件(3D文件)
  show_mask(index_list, image_details,mask_data_not_null, show=True)   #展示mask,暂时不知道怎么搞成3D,有会的帮忙补充一下。

train.csv:

 

展示结果:

生成3D图像:

 

posted @ 2022-05-02 11:55  九叶草  阅读(252)  评论(0编辑  收藏  举报