Построение координат X1, Y1, X2, Y2 с Python с открытым CV - PullRequest
Новая
       5

Построение координат X1, Y1, X2, Y2 с Python с открытым CV

0 голосов
/ 24 марта 2020

В настоящее время я работаю над программой распознавания изображений. Я идентифицировал десять элементов на картинке, и я могу найти координаты тех элементов, которые я обнаружил с помощью алгоритма Mask R-CNN. Теперь я также обнаружил текст рядом с этими элементами (текст, представляющий имя элемента). Так что теперь я должен сопоставить текстовое имя с элементом (в основном через кратчайшее расстояние). Хорошо, вот основной вопрос. На самом деле, основываясь на координатах X1, Y1, X2, Y2, я попробовал CV2.Puttext, но каким-то образом я не могу нанести координаты (по крайней мере, мне достаточно X1, Y1) поверх этих блоков. Кто-то может помочь? 

image = cv2.imread('/content/drive/My Drive/Contest/Image10.png')
image = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
image = imutils.resize(image, width=512)
# perform a forward pass of the network to obtain the results
print("[INFO] making predictions with Mask R-CNN...")
r = model.detect([image], verbose=1)[0]
image = cv2.cvtColor(image, cv2.COLOR_RGB2BGR)
# loop over the predicted scores and class labels
for i in range(0, len(r["scores"])):
    (startY, startX, endY, endX) = r["rois"][i]
    classID = r["class_ids"][i]
    label = CLASS_NAMES[classID]
    score = r["scores"][i]
    color = [int(c) for c in np.array(COLORS[classID]) * 255]
    text = "{}".format(startX, startY)
    y = startY - 10 if startY - 10 > 10 else startY + 10
    cv2.putText(image, text, (startX,startY), cv2.FONT_HERSHEY_SIMPLEX,0.6, color, 2)

введите описание изображения здесь

...