0

我收到了一张包含星星和椭圆的图像,我的任务是检测哪个是哪个,并计算每个图像中包含多少个。一张这样的椭圆形图像看起来像这样

我首先尝试使用 OpenCV 使用诸如this onethis one之类的教程来解决问题。

但是,我似乎在限制椭圆时遇到了问题,一个导致 1 个椭圆的计数,而另一个导致 330 个计数。

然后我尝试使用 YOLOv4,认为它在处理两个不同的类(星形和椭圆形)时会更有用。我在示例图像上使用了顶部尝试绑定框的以下代码。

box, label, count = cv.detect_common_objects(img)
output = draw_bbox(img, box, label, count)
output = cv2.cvtColor(img, cv2.COLOR_BGR2RGB)
plt.figure(figsize= (10, 10))
plt.axis("off")
plt.imshow(img1)
plt.show()

但是我收到了 IndexError: Invalid index to scalar variable。

任何人都可以为我指出如何进行的正确方向吗?我首先需要能够为一个班级做这件事,然后是多个班级,然后再扩展到自动为几个图像做这件事。

谢谢

4

0 回答 0