【发布时间】:2021-10-10 19:59:08
【问题描述】:
我正在尝试使用 keras ImageDataGenerator.flow_from_dataframe() 方法即时生成图像数据,因为我正在处理的数据集太大而无法一次性加载到内存中。
源图像数据文件为DICOM文件,flow_from_dataframe()方法不支持。
是否可以(轻松)扩展 flow_from_dataframe() 以处理 DICOM(或其他不受支持的)图像/输入?
也许可以在每个不受支持的文件上运行自定义预处理函数,返回标准化(窗口化/光度校正)numpy 数组,然后允许 ImageDataGenerator 实例继续进行。
我可以在自己的安装中编辑源代码,但首选可在 vanilla keras 上使用的通用解决方案,以确保可移植到其他平台(尤其是 Kaggle)!!
【问题讨论】:
标签: python keras computer-vision medical-imaging