正在加载包含图像和标签的.npy文件,以便在tf.keras中进行培训

2024-09-28 03:17:35 发布

您现在位置:Python中文网/ 问答频道 /正文

我对tensorflow和keras很陌生。刚刚开始使用TensorFlow2和KerasAPI。我或多或少了解如何建立一个网络和培训它,但我没有´我没有找到任何可以解释如何使用我自己的数据集的东西:

我基本上想做医学图像分割。为此,我将所有患者图像保存在一个.npy文件中。问题是,这个.npy文件包含的东西比图像多,它还包含第二个元素,即实际的分段(标签)和其他包含图像信息的额外条目(如分辨率等)。所以基本上如果我用python加载.npy文件,我有一个数组,它的第一个条目是3D体积(作为np数组)图像,第二个元素是3D分段图像(另一个np数组)。我想用这些来训练。我想我可以在不同的文件中分离它们,然后将路径提供给tf,但是如果可能的话,我想尝试使用数据。 另一件事是,图像保存在三维,如果我想训练一个二维FCN我应该如何继续加载这些三维图像的训练数据

提前谢谢


Tags: 文件数据图像网络元素tensorflownp条目

热门问题