Подсчет объектов на изображениях с использованием OpenCV/YOLOv4 ⇐ Python

[Anonymous]

Мне дали изображение, содержащее звезды и овалы, и мне было поручено определить, что есть что, и подсчитать, сколько каждого из них содержится в изображении. Одно такое изображение с овалами выглядит только так.
Сначала я пытался решить проблему с помощью OpenCV, используя такие руководства, как этот и этот.
Однако я, похоже, столкнулся с проблемами при ограничении овалов: один дает счет 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: неверный индекс скалярной переменной.

Может ли кто-нибудь указать мне правильное направление, как действовать? Сначала мне нужно иметь возможность сделать это для одного класса, а затем для нескольких классов, прежде чем переходить к автоматическому выполнению этого для нескольких изображений.

Подробнее здесь: https://stackoverflow.com/questions/709 ... ncv-yolov4