Невозможно рисовать линии с помощью OpenCV
Я создал функцию per_frame, которая загружается в детектор ImageAI. Я хочу провести линию между центроидами, которые соответствуют критериям расстояния (если расстояние>= find_dist_ratio(x1, y1)). Линии должны быть нарисованы между центроидами всех объектов, соответствующих критериям, я попытался изменить это и, наконец, получил его без ошибок, но линия не отображается в выходном видео. Спасибо за помощь!
def dist_func(counting, output_objects_array,output_objects_count):
a =[]
ret, frame = camera.read()
for d in output_objects_array:
x1 = d['box_points'][0]
y1 = d['box_points'][1]
x2 = d['box_points'][2]
y2 = d['box_points'][3]
centroid = (int((x1 + x2) / 2), int((y1 + y2) / 2))
a.append(centroid)
for i in range(len(a)):
for j in range(i+1, len(a)):
distance = euc_dist(a[i],a[j])
if distance >= find_dist_ratio(x1, y1):
print('close enough')
x, y = a[i]
X, Y = a[j]
cv2.line(frame, (x, y), (X, Y), (255, 0, 0), 5)
1 ответ
Решение
Это может показаться глупым, но в вашем фрагменте кода я не могу понять, действительно ли вы показываете фрейм. И если переменные x и y верны (в нижнем / верхнем регистре)
См. Этот пример из документации:
# Create a black image
img = np.zeros((512,512,3), np.uint8)
# Draw a diagonal blue line with thickness of 5 px
cv.line(img,(0,0),(511,511),(255,0,0),5)
Для отображения нарисованной здесь линии также следует разместить (после рисования)
cv2.imshow("Line draw", img)
Функции рисования в документации