HPF的轻量级封装

非参数Hochbaum伪流（HPF）最小割/最大流算法的Python轻量级封装。原始源代码由Bala Chandran和Dorit S. Hochbaum提供，可在此处找到。在此封装中使用的C++代码已由Patrick M. Jensen重构并发布在此处。

安装

从存储库安装（需要Cython）

git clone https://github.com/Skielex/thinhpf
cd thinhpf
pip install .

该软件包将在稍后提供在PyPI上。

小型示例

import thinhpf

hpf = thinhpf.hpf()

# Add s and t.
next_node_id = hpf.add_node(2)

def offset(n):
    return 2 + n

s = 0
t = 1

hpf.set_source(0)
hpf.set_sink(1)

# Number of nodes to add.
nodes_to_add = 2

# Add two nodes.
next_node_id = hpf.add_node(nodes_to_add)

# Add edges.
hpf.add_edge(s, offset(0), 5)  # s     --5->   n(0)
hpf.add_edge(offset(0), t, 1)  # n(0)  --1->   t
hpf.add_edge(offset(1), t, 3)  # n(1)  --3->   t
hpf.add_edge(offset(0), offset(1), 2)  # n(0)  --2->   n(1)
hpf.add_edge(offset(1), offset(0), 1)  # n(1)  --1->   n(0)
# Find maxflow/cut hpf.
hpf.mincut()
flow = hpf.compute_maxflow()

for n in range(nodes_to_add):
    segment = hpf.what_label(n)
    print('Node %d has label %d.' % (n, segment))
# Node 0 has label 0.
# Node 1 has label 1.

print('Flow: %s' % flow)
# Maximum flow: 3

相关存储库

• GitHub上的Hochbaum Group
• 原始源码网站
• slgbuilder Python软件包（CVPR 2020）
• shrdr Python软件包（ICCV 2021）
• thinqpbo Python软件包
• thinmaxflow Python软件包
• 最大流/最小割算法的C++实现

许可证

Bala Chandran和Dorit S. Hochbaum的原始C代码以及因此的（之前为pseudo.c）内容是根据学术许可证发布的（见LICENSE文件）。有关更多信息，请参阅原始网站。