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)
