使用证据检测图像中对象之间的连接

Question

我正在尝试下图中的跟踪线。所以我的输入图像是这样的：

到目前为止，使用深度学习，我能够区分不同的形状，我得到以下输出，我现在有形状和位置的类型：

这似乎对我很好。此外，我成功找到了如下形状之间的界限，我想用这个作为证据：

可以检测每个形状之间的所有线条。

问题：

我的目标是检测哪个形状/物体与哪个形状以及哪个方向相连。有一段时间我无视指示（但欢迎任何想要发现的想法）。但我真的很困惑我应该如何解决这个问题＆＃34;追踪＆＃34;线。

起初我想到了一种算法，它可以连接＆＃34;两个对象，如果我们找到足够的＆＃34;证据＆＃34;，但在大多数图像中我有一个树状结构，从一个对象有三个或三个以上的连接（在输入图像的第4和第5级之间），让这个简单的方法变得困难。

有人曾经解决过这样的问题吗？我只是在寻找算法，编程语言不是问题，但我目前正在使用python。任何帮助将非常感激。