Обнаружение объекта - медленная обработка видео - PullRequest
Новая
       13

Обнаружение объекта - медленная обработка видео

0 голосов
/ 17 мая 2019

Я запускаю пример кода, который поставляется с моделью обнаружения объектов, я сделал модификацию для чтения видео вместо веб-камеры, проблема в том, что окно открывается и воспроизводит видео, но очень медленно (на самом деле очень медленно , не превышает 1 fps я думаю) 

video = cv2.VideoCapture(PATH_TO_VIDEO)


while(video.isOpened()):

    ret, frame = video.read()
    frame_expanded = np.expand_dims(frame, axis=0)

    # Perform the actual detection by running the model with the image as input
    (boxes, scores, classes, num) = sess.run(
[detection_boxes, detection_scores, detection_classes, num_detections],
    feed_dict={image_tensor: frame_expanded})

    # Draw the results of the detection (aka 'visulaize the results')
    vis_util.visualize_boxes_and_labels_on_image_array(
frame,
np.squeeze(boxes),
np.squeeze(classes).astype(np.int32),
np.squeeze(scores),
category_index,
use_normalized_coordinates=True,
line_thickness=8,
min_score_thresh=0.60)

    # All the results have been drawn on the frame, so it's time to display it.
    cv2.imshow('Object detector', frame)

    # Press 'q' to quit
    if cv2.waitKey(1) == ord('q'):
        break

video.release()
cv2.destroyAllWindows()
...