def __init__(self, browser, regions, parent): QFrame.__init__(self, parent) layout = QGridLayout(self) layout.setContentsMargins(5, 5, 5, 5) self.browser = browser box_detect = QCheckBox('Detect Objects', self) box_detect.setChecked(False) box_detect.toggled.connect(self._on_detect_objects) layout.addWidget(box_detect, 0, 0) self._box_detect = box_detect self._box_region = QComboBox(self) self._box_region.setEnabled(box_detect.checkState() == Qt.Checked) self.update_regionbox(regions) self._box_region.activated[int].connect( self._on_current_region_changed) if len(regions) > 0: self._box_region.setCurrentIndex(0) layout.addWidget(self._box_region, 1, 0, 1, 2) layout.addWidget(QLabel('Show Contours:'), 2, 0) box = QRadioButton('by color', self) box.setEnabled(False) box.setChecked(True) box.toggled.connect(self._on_show_by_color) layout.addWidget(box, 3, 0) self._box_contours = box self._btn_contour_color = ColorButton(None, self) self._btn_contour_color.setEnabled( box_detect.checkState() == Qt.Checked) color = QColor('white') color.setAlphaF(1) self._btn_contour_color.set_color(color) self._btn_contour_color.color_changed.connect( \ self._on_contour_color_changed) layout.addWidget(self._btn_contour_color, 3, 1) box_classify = QRadioButton('by classification', self) box.setEnabled(False) box.setChecked(False) box_classify.toggled.connect(self._on_show_by_classification) box_classify.setChecked(False) layout.addWidget(box_classify, 4, 0) self._box_classify = box_classify box_show_contours = QCheckBox('show all', self) box_show_contours.setChecked(True) box_show_contours.toggled.connect(self._on_toggle_show_contours) box_show_contours.setEnabled(box_detect.checkState() == Qt.Checked) layout.addWidget(box_show_contours, 5, 0) self._box_show_contours = box_show_contours
def __init__(self, parent, browser, settings, imagecontainer): Module.__init__(self, parent, browser) self._current_class = None self._detect_objects = False self._current_scene_items = set() self._settings = settings self._imagecontainer = imagecontainer self._annotations = Annotations() self._object_items = {} splitter = QSplitter(Qt.Vertical, self) grp_box = QGroupBox('Classes', splitter) layout = QBoxLayout(QBoxLayout.TopToBottom, grp_box) layout.setContentsMargins(5, 10, 5, 5) class_table = QTableWidget(grp_box) class_table.setEditTriggers(QTableWidget.NoEditTriggers) class_table.setSelectionMode(QTableWidget.SingleSelection) class_table.setSelectionBehavior(QTableWidget.SelectRows) #class_table.setSortingEnabled(True) class_table.setColumnCount(4) class_table.setHorizontalHeaderLabels(['Name', 'Label', 'Color', 'Samples', ]) class_table.resizeColumnsToContents() class_table.currentItemChanged.connect(self._on_class_changed) class_table.setStyleSheet('font-size: 10px;') layout.addWidget(class_table) self._class_table = class_table frame2 = QFrame(grp_box) layout2 = QBoxLayout(QBoxLayout.LeftToRight, frame2) layout2.setContentsMargins(0,0,0,0) self._import_class_definitions_btn = QPushButton('Import class definitions') layout2.addWidget(self._import_class_definitions_btn) self._import_class_definitions_btn.clicked.connect(self._on_import_class_definitions) layout.addWidget(frame2) frame2 = QFrame(grp_box) layout2 = QBoxLayout(QBoxLayout.LeftToRight, frame2) layout2.setContentsMargins(0,0,0,0) self._class_sbox = QSpinBox(frame2) self._class_color_btn = ColorButton(None, frame2) self._class_sbox.setRange(0, 1000) self._class_text = QLineEdit(frame2) layout2.addWidget(self._class_color_btn) layout2.addWidget(self._class_sbox) layout2.addWidget(self._class_text) layout.addWidget(frame2) frame2 = QFrame(grp_box) layout2 = QBoxLayout(QBoxLayout.LeftToRight, frame2) layout2.setContentsMargins(0,0,0,0) btn = QPushButton('Apply', frame2) btn.clicked.connect(self._on_class_apply) layout2.addWidget(btn) btn = QPushButton('Add', frame2) btn.clicked.connect(self._on_class_add) layout2.addWidget(btn) btn = QPushButton('Remove', frame2) btn.clicked.connect(self._on_class_remove) layout2.addWidget(btn) layout.addWidget(frame2) splitter.addWidget(grp_box) grp_box = QGroupBox('Annotations', splitter) layout = QBoxLayout(QBoxLayout.TopToBottom, grp_box) layout.setContentsMargins(5, 10, 5, 5) ann_table = QTableWidget(grp_box) ann_table.setEditTriggers(QTableWidget.NoEditTriggers) ann_table.setSelectionMode(QTableWidget.SingleSelection) ann_table.setSelectionBehavior(QTableWidget.SelectRows) #ann_table.setSortingEnabled(True) column_names = ['Position', 'Frame', 'Samples'] if self._imagecontainer.has_multiple_plates: column_names = ['Plate'] + column_names ann_table.setColumnCount(len(column_names)) ann_table.setHorizontalHeaderLabels(column_names) ann_table.resizeColumnsToContents() ann_table.currentItemChanged.connect(self._on_anntable_changed) ann_table.setStyleSheet('font-size: 10px;') layout.addWidget(ann_table) self._ann_table = ann_table splitter.addWidget(grp_box) frame = QFrame(grp_box) layout_frame = QBoxLayout(QBoxLayout.LeftToRight, frame) layout_frame.setContentsMargins(0, 0, 0, 0) btn = QPushButton('New', frame) btn.clicked.connect(self._on_new_classifier) layout_frame.addWidget(btn) #layout_frame.addSpacing(5) btn = QPushButton('Open', frame) btn.clicked.connect(self._on_open_classifier) layout_frame.addWidget(btn) btn = QPushButton('Save', frame) btn.clicked.connect(self._on_save_classifier) layout_frame.addWidget(btn) #layout_frame.addSpacing(5) btn = QPushButton('Save as', frame) btn.pressed.connect(self._on_saveas_classifier) layout_frame.addWidget(btn) layout.addWidget(frame) layout = QBoxLayout(QBoxLayout.TopToBottom, self) layout.setContentsMargins(5, 5, 5, 5) layout.addWidget(splitter) self._learner = self._init_new_classifier() self._action_grp = QActionGroup(browser) class_fct = lambda id: lambda : self._on_shortcut_class_selected(id) for x in range(1,11): action = browser.create_action( 'Select Class Label %d' % x, shortcut=QKeySequence(str(x) if x < 10 else '0'), slot=class_fct(x)) self._action_grp.addAction(action) browser.addAction(action) browser.coordinates_changed.connect(self._on_coordinates_changed) browser.show_objects_toggled.connect(self._on_show_objects) browser.show_contours_toggled.connect(self._on_show_contours_toggled)