def train(self):
if self.started:
return
self.worker = MNISTWorker(self)
self.new_data.connect(self.plot)
self.worker.start()
self.started = True
self.update()
self.status_message.emit(str('Training started...'))
class MNISTGUI(QtGui.QFrame):
""" Implement QFrame, which is a subclass of QWidget """
status_message = pyqtSignal(str)
score_message = pyqtSignal(str)
new_data = pyqtSignal(object)
def __init__(self, parent):
QtGui.QFrame.__init__(self, parent)
self.worker = None
self.started = False
pg.setConfigOption('background', 'w')
self.my_plot = pg.PlotWidget()
self.my_plot.getAxis('bottom').setPen('k')
self.my_plot.getAxis('left').setPen('k')
self.hBoxLayout = QtGui.QHBoxLayout()
self.widget_size_px = 600
self.init_ui()
def init_ui(self):
""" Initialize the UI """
self.setFocusPolicy(QtCore.Qt.StrongFocus)
size = QtCore.QSize(self.widget_size_px, self.widget_size_px)
self.setMinimumSize(size)
self.parent().adjustSize()
self.parent().setWindowTitle('Module 5 - MNIST classification')
self.hBoxLayout.setMargin(0)
self.hBoxLayout.addWidget(self.my_plot)
self.setLayout(self.hBoxLayout)
def start_training(self):
""" Start the search in the worker thread """
self.end()
self.train()
def ended(self):
self.status_message.emit('Terminated..')
def end(self):
if not self.started:
return
self.worker.end_worker()
self.worker = None
self.started = False
def train(self):
if self.started:
return
self.worker = MNISTWorker(self)
self.new_data.connect(self.plot)
self.worker.start()
self.started = True
self.update()
self.status_message.emit(str('Training started...'))
def test(self):
if self.started:
return
if not self.worker:
self.status_message.emit(str('Train first...'))
return
self.worker.test()
self.started = True
self.update()
self.status_message.emit(str('Testing started...'))
def plot(self, data):
self.my_plot.plot(data, pen="g")
def resizeEvent(self, e): # pylint: disable=invalid-name
""" Handles widget resize """
size = max(e.size().width(), e.size().height())
self.setMinimumSize(QtCore.QSize(size, size))
self.widget_size_px = size
def reset_size(self):
self.widget_size_px = 600
self.setMinimumSize(QtCore.QSize(self.widget_size_px,
self.widget_size_px))
self.parent().adjustSize()
self.update()
