def run(self):
print("executing")
print(self.entryVideoFolderPath.get())
print(self.entryVideoOutput.get())
videoManager = VideoManager(self.entryVideoFolderPath.get(),
self.entryVideoOutput.get(),
self.entryScriptFilePath.get())
def __init__(self, processor, fps, n=2, fullscreen=False, stack=True):
cv2.namedWindow('Frame', cv2.WINDOW_NORMAL)
if fullscreen:
cv2.setWindowProperty('Frame',
cv2.WND_PROP_FULLSCREEN, cv2.cv.CV_WINDOW_FULLSCREEN)
if stack:
cv2.resizeWindow('Frame', 2133, 600)
else:
cv2.resizeWindow('Frame', 1066, 600)
self.processor = processor
self.videoManager = VideoManager(n)
self.previousImg = None
self.interval = int(1000 / fps)
self.stack = stack
self.screenshot = 0
self.recording = False
from mp4_frames import get_ready_data_dir
from face_detector import MTCNNDetector
from mp4_frames import read_video
from image_sampler import cut_frame
from featureline import _get_face_boxes
from PIL import Image
import numpy as np
import matplotlib.pyplot as plt
v = VideoManager()
df = v._df
target = "plrtfgcwau.mp4"
# target = "wnhlvjdtfg.mp4"
m = df.file == target
if m.sum() == 0:
print("Not registered")
assert m.sum() == 1
def __init__(self, parent=None):
# INIT QT WINDOWS
super(MainWindow, self).__init__(parent=parent)
self.setupUi(self)
# DEFAULT VARIABLES
self.videoManager = VideoManager()
self.meshManager = MeshManager()
self.isDragging = False
self.faceDetector = None
self.progressDialog = QtWidgets.QProgressDialog(
"Loading...", "Stop", 0, 100, self)
self.lastUpdate = 0
self.maxWidth = self.VideoWidget.geometry().width()
self.maxHeight = self.VideoWidget.geometry().height()
self.centerX = self.maxWidth / 2. # center of the video
self.centerY = self.maxHeight / 2.
self.zoom = 1.
self.mousePos = QtCore.QPointF(0, 0)
self.videoFormats = []
self.meshFormats = []
# DISABLED ALL BUTTONS
self.setEnabledButtons(False)
# PUSH BUTTONS EVENTS
self.read_PB.clicked.connect(lambda: self.videoManager.play())
self.pause_PB.clicked.connect(lambda: self.videoManager.pause())
self.nextFrame_PB.clicked.connect(
lambda: self.videoManager.moveFrame(1, self.draw))
self.lastFrame_PB.clicked.connect(
lambda: self.videoManager.moveFrame(-1, self.draw))
self.next10Frame_PB.clicked.connect(
lambda: self.videoManager.moveFrame(10, self.draw))
self.last10Frame_PB.clicked.connect(
lambda: self.videoManager.moveFrame(-10, self.draw))
self.headPosition_PB.clicked.connect(lambda: self.getHeadPosition())
self.processAllVideo_PB.clicked.connect(lambda: self.processAllVideo())
self.coordinates_PB.clicked.connect(lambda: self.drawAxisEvent())
self.square_BT.clicked.connect(lambda: self.drawSquareEvent())
self.screenshot_PB.clicked.connect(lambda: self.screenshotEvent())
self.resetOrientation_PB.clicked.connect(
lambda: self.resetOrientationEvent())
self.resetTSH_PB.clicked.connect(lambda: self.resetTHSEvent())
self.resetBC_PB.clicked.connect(lambda: self.resetBCEvent())
# MENU BAR EVENTS
self.actionOpen_video.triggered.connect(self.selectVideo)
self.actionOpen_model.triggered.connect(self.selectModel)
# SLIDERS EVENTS
self.videoSlider.actionTriggered.connect(lambda: self.sliderChanged())
self.confidenceSlider.actionTriggered.connect(
lambda: self.confidenceChanged())
self.fovySlider.actionTriggered.connect(lambda: self.fovyChanged())
self.yawSlider.actionTriggered.connect(lambda: self.yawChanged())
self.pitchSlider.actionTriggered.connect(lambda: self.pitchChanged())
self.rollSlider.actionTriggered.connect(lambda: self.rollChanged())
self.temperatureSlider.actionTriggered.connect(
lambda: self.temperatureChanged())
self.saturationSlider.actionTriggered.connect(
lambda: self.saturationChanged())
self.hueSlider.actionTriggered.connect(lambda: self.hueChanged())
self.brightnessSlider.actionTriggered.connect(
lambda: self.brightnessChanged())
self.contrastSlider.actionTriggered.connect(
lambda: self.contrastChanged())
# MOUSE EVENTS
self.VideoWidget.wheelEvent = self.wheelEvent
self.VideoWidget.mousePressEvent = self.mousePressEvent
self.VideoWidget.mouseReleaseEvent = self.mouseReleaseEvent
self.VideoWidget.mouseMoveEvent = self.mouseMoveEvent
self.GLWidget.wheelEvent = self.wheelEvent
self.GLWidget.mousePressEvent = self.mousePressEvent
self.GLWidget.mouseReleaseEvent = self.mouseReleaseEvent
self.GLWidget.mouseMoveEvent = self.mouseMoveEvent
# RESIZE EVENT
self.VideoWidget.resizeEvent = self.resizeEvent
self.videoProcessTable.resizeEvent = self.resizeProcessTable
