sad

tensorflow的静态图属性导致在模型中不能使用其他的代码添加到模型中,所以只能自己实现在分割图中打框的操作。本来opencv是有的,效果不错,运行效率也很高。

直接使用opencv findContours 和 minAreaRect 方法从分割结果获取box

image

造车

由于模型中间层不能使用opencv,所以自己写了个findContours,效果如下图

从分割图中分离每一个物体 image

image

使用PCA求出分离出物体的主轴,然后画矩形

效果

image

​ 过滤掉太小的目标

image

运行效率,由于是python写的,自己测试测运行时间是opencv的两倍