import os from PyQt4 import QtGui, uic from inference_viewer import * class InferenceControl(QtGui.QMainWindow): def __init__(self, parent=None): super(InferenceControl, self).__init__(parent) self.runningState = False self.initUI() def initUI(self): uic.loadUi("inference_control.ui", self) #self.setStyleSheet("background-color: white") self.upload_comboBox.activated.connect(self.fromFile) self.file_pushButton.clicked.connect(self.browseFile) self.output_pushButton.clicked.connect(self.browseOutput) self.label_pushButton.clicked.connect(self.browseLabel) self.image_pushButton.clicked.connect(self.browseImage) self.val_pushButton.clicked.connect(self.browseVal) self.hier_pushButton.clicked.connect(self.browseHier) self.run_pushButton.clicked.connect(self.runConfig) self.file_lineEdit.textChanged.connect(self.checkInput) self.name_lineEdit.textChanged.connect(self.checkInput) self.idims_lineEdit.textChanged.connect(self.checkInput) self.odims_lineEdit.textChanged.connect(self.checkInput) self.output_lineEdit.textChanged.connect(self.checkInput) self.label_lineEdit.textChanged.connect(self.checkInput) self.image_lineEdit.textChanged.connect(self.checkInput) self.image_lineEdit.textChanged.connect(self.checkInput) self.close_pushButton.clicked.connect(self.closeEvent) self.file_lineEdit.setPlaceholderText("File Directory [required]") self.name_lineEdit.setPlaceholderText("Model Name [required]") self.idims_lineEdit.setPlaceholderText("c,h,w [required]") self.odims_lineEdit.setPlaceholderText("c,h,w [required]") self.padd_lineEdit.setPlaceholderText("r,g,b [optional]") self.pmul_lineEdit.setPlaceholderText("r,g,b [optional]") self.output_lineEdit.setPlaceholderText("Output Directory [required]") self.label_lineEdit.setPlaceholderText("Label File [required]") self.image_lineEdit.setPlaceholderText("Image Folder [required]") self.val_lineEdit.setPlaceholderText("[optional]") self.hier_lineEdit.setPlaceholderText("[optional]") self.close_pushButton.setStyleSheet("color: white; background-color: darkRed") self.gui_checkBox.setChecked(True) self.readSetupFile() def browseFile(self): if self.format_comboBox.currentText() == 'nnef': self.file_lineEdit.setText(QtGui.QFileDialog.getExistingDirectory(self, 'Open Folder', './')) else: self.file_lineEdit.setText(QtGui.QFileDialog.getOpenFileName(self, 'Open File', './', '*')) def browseOutput(self): self.output_lineEdit.setText(QtGui.QFileDialog.getExistingDirectory(self, 'Open Folder', './')) def browseLabel(self): self.label_lineEdit.setText(QtGui.QFileDialog.getOpenFileName(self, 'Open File', './', '*.txt')) def browseImage(self): self.image_lineEdit.setText(QtGui.QFileDialog.getExistingDirectory(self, 'Open Folder', './')) def browseVal(self): self.val_lineEdit.setText(QtGui.QFileDialog.getOpenFileName(self, 'Open File', './', '*.txt')) def browseHier(self): self.hier_lineEdit.setText(QtGui.QFileDialog.getOpenFileName(self, 'Open File', './', '*.csv')) def readSetupFile(self): setupDir = '~/.mivisionx-validation-tool' analyzerDir = os.path.expanduser(setupDir) if os.path.isfile(analyzerDir + "/setupFile.txt"): for line in open(analyzerDir + "/setupFile.txt", "r"): if len(line) > 1: token = line.split(';') if len(token) > 1: modelName = token[1] self.upload_comboBox.addItem(modelName) def fromFile(self): if self.upload_comboBox.currentIndex() == 0: self.name_lineEdit.setEnabled(True) self.file_lineEdit.setEnabled(True) self.batch_comboBox.setEnabled(True) self.mode_comboBox.setCurrentIndex(0) self.idims_lineEdit.setEnabled(True) self.odims_lineEdit.setEnabled(True) self.label_lineEdit.setEnabled(True) self.output_lineEdit.setEnabled(True) self.image_lineEdit.setEnabled(True) self.val_lineEdit.setEnabled(True) self.hier_lineEdit.setEnabled(True) self.padd_lineEdit.setEnabled(True) self.pmul_lineEdit.setEnabled(True) self.gui_checkBox.setEnabled(True) self.fp16_checkBox.setEnabled(True) self.replace_checkBox.setEnabled(True) self.verbose_checkBox.setEnabled(True) self.loop_checkBox.setEnabled(True) self.file_pushButton.setEnabled(True) self.format_comboBox.setEnabled(True) self.output_pushButton.setEnabled(True) self.label_pushButton.setEnabled(True) self.image_pushButton.setEnabled(True) self.val_pushButton.setEnabled(True) self.hier_pushButton.setEnabled(True) self.format_comboBox.setCurrentIndex(0) self.name_lineEdit.clear() self.file_lineEdit.clear() self.idims_lineEdit.clear() self.odims_lineEdit.clear() self.label_lineEdit.clear() self.output_lineEdit.clear() self.image_lineEdit.clear() self.val_lineEdit.clear() self.hier_lineEdit.clear() self.padd_lineEdit.clear() self.pmul_lineEdit.clear() self.gui_checkBox.setChecked(True) self.fp16_checkBox.setChecked(False) self.replace_checkBox.setChecked(False) self.verbose_checkBox.setChecked(False) self.loop_checkBox.setChecked(True) else: modelName = self.upload_comboBox.currentText() setupDir = '~/.mivisionx-validation-tool' analyzerDir = os.path.expanduser(setupDir) for line in open(analyzerDir + "/setupFile.txt", "r"): tokens = line.split(';') if len(tokens) > 1: name = tokens[1] if modelName == name: if tokens[0] == 'caffe': format = 0 elif tokens[0] == 'onnx': format = 1 else: format = 2 self.format_comboBox.setCurrentIndex(format) self.name_lineEdit.setText(tokens[1]) self.file_lineEdit.setText(tokens[2]) batch_index = 0 if tokens[3] == '64' else 1 self.batch_comboBox.setCurrentIndex(batch_index) self.idims_lineEdit.setText(tokens[4]) self.odims_lineEdit.setText(tokens[5]) self.label_lineEdit.setText(tokens[6]) self.output_lineEdit.setText(tokens[7]) self.image_lineEdit.setText(tokens[8]) self.val_lineEdit.setText(tokens[9]) self.hier_lineEdit.setText(tokens[10]) self.padd_lineEdit.setText(tokens[11]) self.pmul_lineEdit.setText(tokens[12]) self.fp16_checkBox.setChecked(True) if tokens[13] == 'yes\n' or tokens[13] == 'yes' else self.fp16_checkBox.setChecked(False) self.replace_checkBox.setChecked(True) if tokens[14] == 'yes\n' or tokens[14] == 'yes' else self.replace_checkBox.setChecked(False) self.verbose_checkBox.setChecked(True) if tokens[15] == 'yes\n' or tokens[15] == 'yes' else self.verbose_checkBox.setChecked(False) self.loop_checkBox.setChecked(True) if tokens[16] == 'yes\n' or tokens[16] == 'yes' else self.replace_checkBox.setChecked(False) self.name_lineEdit.setEnabled(False) self.file_lineEdit.setEnabled(False) self.batch_comboBox.setEnabled(False) self.idims_lineEdit.setEnabled(False) self.odims_lineEdit.setEnabled(False) self.label_lineEdit.setEnabled(False) self.output_lineEdit.setEnabled(False) self.image_lineEdit.setEnabled(False) self.val_lineEdit.setEnabled(False) self.hier_lineEdit.setEnabled(False) self.padd_lineEdit.setEnabled(False) self.pmul_lineEdit.setEnabled(False) self.fp16_checkBox.setEnabled(False) self.output_pushButton.setEnabled(False) self.label_pushButton.setEnabled(False) self.image_pushButton.setEnabled(False) self.val_pushButton.setEnabled(False) self.hier_pushButton.setEnabled(False) self.verbose_checkBox.setEnabled(False) self.file_pushButton.setEnabled(False) self.format_comboBox.setEnabled(False) def checkInput(self): if not self.file_lineEdit.text().isEmpty() and not self.name_lineEdit.text().isEmpty() \ and not self.idims_lineEdit.text().isEmpty() \ and not self.odims_lineEdit.text().isEmpty() and not self.output_lineEdit.text().isEmpty() \ and not self.label_lineEdit.text().isEmpty() and not self.image_lineEdit.text().isEmpty(): self.run_pushButton.setEnabled(True) self.run_pushButton.setStyleSheet("background-color: lightgreen") else: self.run_pushButton.setEnabled(False) self.run_pushButton.setStyleSheet("background-color: 0") def runConfig(self): model_format = (str)(self.format_comboBox.currentText()) model_name = (str)(self.name_lineEdit.text()) model_location = (str)(self.file_lineEdit.text()) batch_size = (str)(self.batch_comboBox.currentText()) rali_mode = self.mode_comboBox.currentIndex() + 1 input_dims = (str)('%s' % (self.idims_lineEdit.text())) output_dims = (str)('%s' % (self.odims_lineEdit.text())) label = (str)(self.label_lineEdit.text()) output_dir = (str)(self.output_lineEdit.text()) image_dir = (str)(self.image_lineEdit.text()) image_val = (str)(self.val_lineEdit.text()) hierarchy = (str)(self.hier_lineEdit.text()) if len(self.padd_lineEdit.text()) < 1: add = '[0,0,0]' else: add = (str)('[%s]' % (self.padd_lineEdit.text())) if len(self.pmul_lineEdit.text()) < 1: multiply = '[1,1,1]' else: multiply = (str)('[%s]' % (self.pmul_lineEdit.text())) gui = 'yes' if self.gui_checkBox.isChecked() else 'no' fp16 = 'yes' if self.fp16_checkBox.isChecked() else 'no' replace = 'yes' if self.replace_checkBox.isChecked() else 'no' verbose = 'yes' if self.verbose_checkBox.isChecked() else 'no' loop = 'yes' if self.loop_checkBox.isChecked() else 'no' container_logo = self.container_comboBox.currentIndex() fps_file = '' cpu_name = self.cpu_comboBox.currentText() gpu_name = self.gpu_comboBox.currentText() self.runningState = True self.close() viewer = InferenceViewer(model_name, model_format, image_dir, model_location, label, hierarchy, image_val, input_dims, output_dims, batch_size, output_dir, add, multiply, verbose, fp16, replace, loop, rali_mode, gui, container_logo, fps_file, cpu_name, gpu_name, self) if gui == 'yes': #viewer.show() viewer.showMaximized() def closeEvent(self, event): if not self.runningState: exit(0)