条件随机场(CRF)和稠密CRF的开源工具包
SimpleCRF的Python项目详细描述
简单的
Matlab和Python wrap of Conditional Random Field(CRF)和全连通(稠密)CRF用于二维和三维图像分割,根据以下论文:
[1]Yuri Boykov和Vladimir Kolmogorov,“视觉中能量最小化最小切/最大流算法的实验比较”,IEEE TPAMI,2004。在
[2]Philipp Krähenbühl和Vladlen Koltun,“具有高斯边势的全连接CRF的有效推理”,NIPS,2011。在
[3]Kamnitsas等人在“高效多尺度3D CNN与全连接CRF进行精确脑损伤分割”,医学图像分析,2017年。在
依赖性
安装
安装人:^{
在} 或者,可以通过以下两个步骤编译源文件:
在
python setup.py build python setup.py install
示例
使用此软件包的一些演示包括:
在examples/demo_maxflow.py
:使用maxflow自动交互式分割2D和3D图像。在
在examples/demo_densecrf.py
:使用稠密CRF进行二维灰度和RGB图像分割。在
在examples/demo_densecrf3d.py
:使用3D密集CRF进行三维多模态图像分割。在
模块
maxflow
有以下四个功能。请注意,当前版本只支持二进制分段。在
在maxflow.maxflow2d()
用于二维自动分割。在
在maxflow.interactive_maxflow2d()
用于二维交互式分割。在
在maxflow.maxflow3d()
用于三维自动分割。在
在maxflow.interactive_maxflow3d()
用于三维交互式分割。在
denseCRF
具有以下功能。它可以处理多类分割,并且只支持RGB图像。在
denseCRF.densecrf()
用于二维自动分割。在
denseCRF3D
具有以下功能。它可以处理多类分割问题。输入通道号可以是1-5。在
denseCRF3D.densecrf3d()
用于三维自动分割。在
- 项目
标签: