Видео выхода Yolo Darkflow не приходит - PullRequest
Новая
       83

Видео выхода Yolo Darkflow не приходит

0 голосов
/ 18 апреля 2020

Используя команду потока Flowflow, я могу создать новое видео с ограничительной рамкой, которое я сохраняю, используя команду 

flow --model cfg/neww_yolo.cfg --load bin/yolov2.weights --demo car.mp4 --saveVideo

Но когда я запускаю сценарий python для обработки видео, я получаю количество FPS на выходе, но приложение открывается, и когда я указываю курсор мыши, он не отвечает. Мои python сценарии: 

import cv2
from darkflow.net.build import TFNet
import numpy as np
import time

option = {
    'model' : 'cfg/neww_yolo.cfg',
    'load' : 'bin/yolov2.weights',
    'threshold' : 0.15,
    'demo' : 'car.mp4'
}
tfnet = TFNet(option)
capture = cv2.VideoCapture('car.mp4') 
colors = [tuple(255* np.random.rand(3)) for i in range(5)]
for color in colors:
    print(color)
while(capture.isOpened()):
    stime = time.time()
    ret , frame = capture.read()
    frame = np.asarray(frame) 
#     print(frame)
    if ret:
        results =tfnet.return_predict(frame)
        for color, result in zip(colors,results):
            tl = (result['topleft']['x'], result['topleft']['y'])
            br = (result['bottomright']['x'], result['bottomright']['y'])
            label = result['label']
            frame = cv2.rectangle(frame, tl, br, color, 7)
            frame = cv2.putText(frame, label,tl , cv2.FONT_HERSHEY_COMPLEX, 1, (0,0,0),2)   
        cv2.imshow('frame',frame)
        print('FPS{:.1f}'.format(1 / (time.time() - stime)))
        if 0xFF == ord('q'):
            break
    else:
        capture.release()
        cv2.destroyAllWindows()
        break

В выводе я получаю 

FPS0.7
FPS0.7
FPS0.7
FPS0.7
FPS0.7
FPS0.7
FPS0.7
FPS0.7
... and so on till many iteration

Таким образом, я использую macOS, и в фоновом режиме приложение открыто и зависает над ним, скажем, не отвечает , Так есть ли способ начать видео с ограничительной рамки или любым другим способом, как в потоке, из которого я могу сохранить видео файл с расширением .avi, используя этот скрипт.

...