ImageMagick 或其他方法：如何在图像中找到对象中心的坐标？

Question

我正在尝试在图像（杯子）中找到对象的坐标。

我最终需要得出的信息是 x、y、宽度和高度 - 其中 x 和 y 应该是包含杯子（而不是左上角）的矩形的中心，宽度和高度应该是相对于图像宽度和高度的浮点值（从 0.0 到 1.0）。

我尝试安装 ImageMagick，加载图像，然后在 cmd 中运行以下命令：

magick image1.jpg -define connected-components:verbose=true -connected-components 8 -auto-level image1.jpg

我收到错误：ma​​gick: too many objects `image1.jpg' @ error/vision.c/ConnectedComponentsImage/437。

有谁知道如何解决这个问题，或者任何其他相对简单的方法来找到所需的坐标？

Answer 1

很遗憾，您的对象与背景颜色相同。所以通过简单的图像处理很难做到这一点。您需要一种 AI/深度学习方法，例如 http://remove.bg

所以我使用了http://remove.bg 来执行此操作，但请注意 remove.bg 将图像缩小了一点。

从中我提取了 alpha 通道：

convert mug.png -alpha extract mug_alpha.png

WxH=`convert mug_alpha.png -format "%wx%h" info:`
W=`echo $WxH | cut -dx -f1`
H=`echo $WxH | cut -dx -f2`

bbox=`convert mug_alpha.png \
-type bilevel \
-define connected-components:mean-color=true \
-define connected-components:exclude-header=true \
-define connected-components:verbose=true \
-define connected-components:keep-top=1 \
-connected-components 4 \
null: | grep "gray(255)" | awk '{print $2}' | tr "x" "+"`
wd=`echo $bbox | cut -d+ -f1`
ht=`echo $bbox | cut -d+ -f2`
xoff=`echo $bbox | cut -d+ -f3`
yoff=`echo $bbox | cut -d+ -f4`

xcenter=`convert xc: -format "%[fx:$xoff + $wd/2]" info:`
ycenter=`convert xc: -format "%[fx:$yoff + $ht/2]" info:`
width=`convert xc: -format "%[fx:$wd/$W]" info:`
height=`convert xc: -format "%[fx:$ht/$H]" info:`
echo "xcenter=$xcenter, ycenter=$ycenter, width=$width, height=$height"

xcenter=277, ycenter=260, width=0.563433, height=0.60515

left=$xoff
top=$yoff
right=$((xoff+wd-1))
bottom=$((yoff+ht-1))
convert mug.png -fill none -stroke red -draw "rectangle $left,$top $right,$bottom" -alpha off mug_box.png
convert mug_alpha.png -fill none -stroke red -draw "rectangle $left,$top $right,$bottom" -alpha off mug_alpha_box.png