def setupUi(self, Dialog):
Dialog.setObjectName("Dialog")
Dialog.resize(733, 628)
self.buttonBox = QtGui.QDialogButtonBox(Dialog)
self.buttonBox.setGeometry(QtCore.QRect(100, 570, 341, 32))
self.buttonBox.setOrientation(QtCore.Qt.Horizontal)
self.buttonBox.setStandardButtons(QtGui.QDialogButtonBox.Cancel
| QtGui.QDialogButtonBox.Ok)
self.buttonBox.setObjectName("buttonBox")
self.video_widget = VideoWidget(Dialog)
self.video_widget.setGeometry(QtCore.QRect(20, 20, 648, 486))
self.video_widget.setObjectName("video_widget")
self.video_widget.push_frame(self._frame)
text = "W: %f<br />H: %f" % (self.w, self.h)
self.measure_text = QtGui.QTextEdit(text, Dialog)
self.measure_text.setGeometry(QtCore.QRect(540, 550, 151, 61))
self.measure_text.setObjectName("measure_text")
self.label = QtGui.QLabel(Dialog)
self.label.setGeometry(QtCore.QRect(540, 520, 55, 18))
self.label.setObjectName("label")
self.label_2 = QtGui.QLabel(Dialog)
self.label_2.setGeometry(QtCore.QRect(270, 540, 171, 21))
self.label_2.setObjectName("label_2")
self.retranslateUi(Dialog)
self.buttonBox.accepted.connect(Dialog.accept)
self.buttonBox.rejected.connect(Dialog.reject)
QtCore.QMetaObject.connectSlotsByName(Dialog)
def __init__(self, expression_to_neutral=False):
super().__init__()
self.expression_to_neutral = expression_to_neutral
if self.expression_to_neutral:
self.title = "From Expression to Neutral"
else:
self.title = "3DMM"
self.setWindowIcon(QtGui.QIcon('images/1UP.ico'))
self.setFixedSize(1200, 700)
self.camera = Camera()
self.progress_bar = QProgressBar(self)
self.count = 0
self.toolbar = self.addToolBar('Main Window')
# self.toolbar_emotions = self.addToolBar('Emotions')
self.expression = "neutral"
self.picture_taken = False
self.bool_onclick = None
self.portrait = None
self.model = Model(None, self.expression, self,
self.expression_to_neutral)
self.video_widget = VideoWidget(self.camera)
self.right_label = RightLabel(self, self.model)
self.main_widget = MainWidget(self.video_widget, self.right_label,
self.camera)
self.setCentralWidget(self.main_widget)
self.model.progress_bar.connect(self.set_progress_bar,
type=Qt.QueuedConnection)
self.combo_box = QComboBox(self)
self.initUI()
def init_ui(self):
self.setWindowTitle('LabelTool')
self.center_window(1200, 800)
self.init_menubar()
self.init_statusbar()
self.video_widget = VideoWidget(max_fps=50)
self.annotation_widget = AnnotationWidget()
self.hbox_layout = QHBoxLayout()
self.hbox_layout.addWidget(self.video_widget, 3)
self.hbox_layout.addWidget(self.annotation_widget, 1)
self.central_widget = QWidget(self)
self.central_widget.setLayout(self.hbox_layout)
self.setCentralWidget(self.central_widget)
def __init__(self, camera, main=False):
super().__init__()
self.title = "3DMM"
self.main = main
# self.setWindowIcon(QtGui.QIcon('images/1UP.ico'))
self.camera = camera
if main:
# self.camera = Camera(1, main, )
self.showFullScreen()
else:
# self.camera = Camera(0)
self.setGeometry(100, 100, 600, 500)
self.video_widget = VideoWidget(self.camera)
self.initUI()
def setupUi(self, MainWindow):
MainWindow.setObjectName(_fromUtf8("MainWindow"))
MainWindow.resize(800, 600)
self.centralWidget = QtGui.QWidget(MainWindow)
self.centralWidget.setObjectName(_fromUtf8("centralWidget"))
self.gridLayout = QtGui.QGridLayout(self.centralWidget)
self.gridLayout.setObjectName(_fromUtf8("gridLayout"))
self.groupBox = QtGui.QGroupBox(self.centralWidget)
self.groupBox.setObjectName(_fromUtf8("groupBox"))
self.verticalLayout = QtGui.QVBoxLayout(self.groupBox)
self.verticalLayout.setObjectName(_fromUtf8("verticalLayout"))
self.video_widget = VideoWidget(self.groupBox)
sizePolicy = QtGui.QSizePolicy(QtGui.QSizePolicy.Fixed,
QtGui.QSizePolicy.Fixed)
sizePolicy.setHorizontalStretch(0)
sizePolicy.setVerticalStretch(0)
sizePolicy.setHeightForWidth(
self.video_widget.sizePolicy().hasHeightForWidth())
self.video_widget.setSizePolicy(sizePolicy)
self.video_widget.setMinimumSize(QtCore.QSize(640, 480))
self.video_widget.setBaseSize(QtCore.QSize(0, 0))
self.video_widget.setObjectName(_fromUtf8("video_widget"))
self.verticalLayout.addWidget(self.video_widget)
self.gridLayout.addWidget(self.groupBox, 0, 0, 1, 1)
self.groupBox_2 = QtGui.QGroupBox(self.centralWidget)
self.groupBox_2.setMinimumSize(QtCore.QSize(200, 0))
self.groupBox_2.setBaseSize(QtCore.QSize(0, 0))
self.groupBox_2.setObjectName(_fromUtf8("groupBox_2"))
self.class_name = QtGui.QTextEdit(self.groupBox_2)
self.class_name.setGeometry(QtCore.QRect(10, 50, 161, 21))
self.class_name.setObjectName(_fromUtf8("class_name"))
self.label = QtGui.QLabel(self.groupBox_2)
self.label.setGeometry(QtCore.QRect(10, 30, 81, 16))
self.label.setObjectName(_fromUtf8("label"))
self.show_mask = QtGui.QCheckBox(self.groupBox_2)
self.show_mask.setGeometry(QtCore.QRect(10, 90, 101, 21))
self.show_mask.setObjectName(_fromUtf8("show_mask"))
self.threshold = QtGui.QSpinBox(self.groupBox_2)
self.threshold.setGeometry(QtCore.QRect(10, 130, 52, 28))
self.threshold.setMaximum(400)
self.threshold.setProperty("value", 80)
self.threshold.setObjectName(_fromUtf8("threshold"))
self.label_2 = QtGui.QLabel(self.groupBox_2)
self.label_2.setGeometry(QtCore.QRect(70, 130, 91, 21))
self.label_2.setObjectName(_fromUtf8("label_2"))
self.gridLayout.addWidget(self.groupBox_2, 0, 1, 1, 1)
self.snap_button = QtGui.QPushButton(self.centralWidget)
self.snap_button.setObjectName(_fromUtf8("snap_button"))
self.gridLayout.addWidget(self.snap_button, 1, 0, 1, 1)
MainWindow.setCentralWidget(self.centralWidget)
self.menuBar = QtGui.QMenuBar(MainWindow)
self.menuBar.setGeometry(QtCore.QRect(0, 0, 800, 26))
self.menuBar.setObjectName(_fromUtf8("menuBar"))
MainWindow.setMenuBar(self.menuBar)
self.mainToolBar = QtGui.QToolBar(MainWindow)
self.mainToolBar.setObjectName(_fromUtf8("mainToolBar"))
MainWindow.addToolBar(QtCore.Qt.TopToolBarArea, self.mainToolBar)
self.statusBar = QtGui.QStatusBar(MainWindow)
self.statusBar.setObjectName(_fromUtf8("statusBar"))
MainWindow.setStatusBar(self.statusBar)
self.mainToolBar.addSeparator()
self.retranslateUi(MainWindow)
QtCore.QMetaObject.connectSlotsByName(MainWindow)
