Revision history [back]

i am trying to save the video with detected objects,it saves the video but dont show any detected object. CODE is import cv2 from darkflow.net.build import TFNet import numpy as np import time import os from PyQt5 import QtCore, QtWidgets, QtGui, uic from PyQt5.QtWidgets import QPushButton, QInputDialog, QLineEdit from PyQt5.uic import loadUi import sys from PIL import Image

option= { 'model':'cfg/yolo.cfg', 'load':'bin/yolov2.weights', 'threshold': 0.3, } class MyWindow(QtWidgets.QMainWindow):

def __init__(self):
    super(MyWindow, self).__init__()
    loadUi("new.ui", self)
    self.testcam.clicked.connect(self.CameraTest)
    self.start.clicked.connect(self.Camera)


def CameraTest(self):
    cap = cv2.VideoCapture(0)
    cap.set(3,640) # set Width
    cap.set(4,480) # set Height
    while(True):
        ret, frame = cap.read()
        gray = cv2.cvtColor(frame, cv2.COLOR_BGR2GRAY)

        cv2.imshow('CamTest', frame)

        k = cv2.waitKey(30) & 0xff
        if k == 27: # press 'ESC' to quit
            break
    cap.release()
    cv2.destroyAllWindows()
def Camera(self):

    tfnet = TFNet(option)
    colors = [tuple(255 * np.random.rand(3)) for _ in range(10)]

    cap= cv2.VideoCapture(0)
    #cap.set(cv2.CAP_PROP_FRAME_WIDTH, 512)
    #cap.set(cv2.CAP_PROP_FRAME_HEIGHT, 512)
    fourcc=cv2.VideoWriter_fourcc(*'XVID')
    out=cv2.VideoWriter('new.avi',fourcc,20.0,(640,480))

    while True:
        stime = time.time()
        ret, frame = cap.read()
        out.write(frame)
        if (ret==True):

            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']
                confidence = result['confidence']
                text = '{}: {:.0f}%'.format(label, confidence * 100)
                frame = cv2.rectangle(frame, tl, br, color, 5)
                frame = cv2.putText(
                    frame, text, tl, cv2.FONT_HERSHEY_COMPLEX, 1, (0, 0, 0), 2)
            cv2.imshow('frame', frame)
            print('FPS {:.1f}'.format(1 / (time.time() - stime)))
            if cv2.waitKey(1) & 0xFF == ord('q'):
                break
        else:
            break

    cap.release()
    out.release()
    cv2.destroyAllWindows()

if __name__ == '__main__': import sys app = QtWidgets.QApplication(sys.argv) window = MyWindow() window.show() sys.exit(app.exec_())