PixelLib 无法正确检测对象

CLR*_*CLR 5 image-processing feature-extraction image-segmentation data-science pixellib

我正在使用的库

import pixellib
from pixellib.instance import instance_segmentation
import cv2
import matplotlib.pyplot as plt
Run Code Online (Sandbox Code Playgroud)

剧本:

segment_image = instance_segmentation()
segment_image.load_model('mask_rcnn_coco.h5')
segmask, output = segment_image.segmentImage("images\example2.jpeg", show_bboxes = True)
cv2.imwrite("exampleoutput.jpeg", output)
print(output.shape)
Run Code Online (Sandbox Code Playgroud)

我不明白为什么它不能突出显示图像的不同部分。

这是我的输出:

这是我的输出

我研究了其他人如何使用 Pixellib,它与他们的完美配合。

我期待的输出:

我期待的输出

Ars*_*rsh 0

升级到 Pixellib 的最新版本。

pip install pixellib --upgrade
Run Code Online (Sandbox Code Playgroud)

并安装tensorflow版本2.5.0

pip install tensorflow==2.5.0
Run Code Online (Sandbox Code Playgroud)

这样做对我来说是固定的。