def __init__(self, modelFile=None, weightsFile=None, solverFile=None, directory=None):
"""
The top-level GUI object, containing three main tabs:
Configure: for graphically editing a network backed by a caffe prototxt file
Train: for monitoring the training process and plotting the accuracies
Test: for visualizing the activations at different points in a (hopefully) trained network
:param modelFile: str
The .prototxt file defining the network architecture
:param weightsFile: str
The .caffemodel file containing the learned network weights
:param solverFile: str
The .prototxt file defining the solver parameters
:param directory: str
The base directory containing the files, for relative file paths
:return:
"""
if directory is not None:
modelFile = directory + modelFile
weightsFile = directory + weightsFile
solverFile = directory + solverFile
self.directory = directory
self.plots = {}
self.plotNodes = {}
self.plotDocks = {}
self.trainLossPlots = {}
QtGui.QMainWindow.__init__(self)
self.resize(1200, 800)
self.setWindowTitle("caffeViz")
self.tabWidget = QtGui.QTabWidget()
self.tabWidget.setSizePolicy(QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding))
self.ui = MView.Ui_tabWidget()
self.ui.setupUi(self.tabWidget)
cw = QtGui.QWidget()
cw.setSizePolicy(QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding))
layout = QtGui.QGridLayout()
cw.setLayout(layout)
layout.addWidget(self.tabWidget)
self.setCentralWidget(cw)
self.fc = NetFlowchart(prototxt=modelFile, weights=weightsFile)
self.fc.sigChartChanged.connect(self.fc.chartNodeEdited)
w = self.fc.widget()
self.ui.splitter.addWidget(self.ui.displayLayout)
# self.plotLabel = QtGui.QLabel()
# self.plotLabelDock = dockarea.Dock('label')
# self.plotLabelDock.addWidget(self.plotLabel)
# self.ui.plotWidget.addDock(self.plotLabelDock)
# self.hoverDock.addWidget(self.hoverText)
# self.addDock(self.hoverDock, 'bottom')
# self.ui.plotWidget.addDock
ctrlList = w.ui.ctrlList
ctrlList.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
w.ui.ctrlList = ctrlList
# w = w._nodes['LayerNode.0'].ctrlWidget()
w.sizePolicy().setHorizontalPolicy(QtGui.QSizePolicy.MinimumExpanding)
self.tabWidget.currentChanged.connect(self.tabChanged)
self.tabWidget.setCurrentIndex(1)
self.tabWidget.setCurrentIndex(2)
self.fc.sigDisplayNodeAdded.connect(self.addPlot)
self.displayControlDock = pg.dockarea.Dock("Control")
formWidget = pg.QtGui.QWidget()
displayCtrlUi = DispCtrl.Ui_displayControlWidget()
displayCtrlUi.setupUi(formWidget)
displayCtrlUi.previousBtn.clicked.connect(self.updateDisplay)
displayCtrlUi.updateBtn.clicked.connect(self.updateDisplay)
displayCtrlUi.nextBtn.clicked.connect(self.updateDisplay)
self.displayCtrlUi = displayCtrlUi
self.displayControlDock.addWidget(formWidget)
self.ui.displayArea.addDock(self.displayControlDock, position="top")
self.solverNode = SolverNode("solver", solverFile, weightsFile, directory)
self.solverNode.sigOutputDataChanged.connect(self.updateTrainPlots)
self.ui.trainButton.pressed.connect(self.solverNode.trainNet)
self.ui.stopButton.pressed.connect(self.solverNode.stopTraining)
self.ui.resumeButton.pressed.connect(self.solverNode.resumeTraining)
self.ui.loadSolverButton.pressed.connect(self.loadSolver)
self.ui.saveSolverButton.pressed.connect(self.saveSolver)
self.ui.trainParamTree.addParameters(self.solverNode.param)
self.trainPlotItem = self.ui.trainGraphicsLayout.plotItem
class MainView(QtGui.QMainWindow):
"""The main view of the application"""
def __init__(self, modelFile=None, weightsFile=None, solverFile=None, directory=None):
"""
The top-level GUI object, containing three main tabs:
Configure: for graphically editing a network backed by a caffe prototxt file
Train: for monitoring the training process and plotting the accuracies
Test: for visualizing the activations at different points in a (hopefully) trained network
:param modelFile: str
The .prototxt file defining the network architecture
:param weightsFile: str
The .caffemodel file containing the learned network weights
:param solverFile: str
The .prototxt file defining the solver parameters
:param directory: str
The base directory containing the files, for relative file paths
:return:
"""
if directory is not None:
modelFile = directory + modelFile
weightsFile = directory + weightsFile
solverFile = directory + solverFile
self.directory = directory
self.plots = {}
self.plotNodes = {}
self.plotDocks = {}
self.trainLossPlots = {}
QtGui.QMainWindow.__init__(self)
self.resize(1200, 800)
self.setWindowTitle("caffeViz")
self.tabWidget = QtGui.QTabWidget()
self.tabWidget.setSizePolicy(QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding))
self.ui = MView.Ui_tabWidget()
self.ui.setupUi(self.tabWidget)
cw = QtGui.QWidget()
cw.setSizePolicy(QtGui.QSizePolicy(QtGui.QSizePolicy.Expanding, QtGui.QSizePolicy.Expanding))
layout = QtGui.QGridLayout()
cw.setLayout(layout)
layout.addWidget(self.tabWidget)
self.setCentralWidget(cw)
self.fc = NetFlowchart(prototxt=modelFile, weights=weightsFile)
self.fc.sigChartChanged.connect(self.fc.chartNodeEdited)
w = self.fc.widget()
self.ui.splitter.addWidget(self.ui.displayLayout)
# self.plotLabel = QtGui.QLabel()
# self.plotLabelDock = dockarea.Dock('label')
# self.plotLabelDock.addWidget(self.plotLabel)
# self.ui.plotWidget.addDock(self.plotLabelDock)
# self.hoverDock.addWidget(self.hoverText)
# self.addDock(self.hoverDock, 'bottom')
# self.ui.plotWidget.addDock
ctrlList = w.ui.ctrlList
ctrlList.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
w.ui.ctrlList = ctrlList
# w = w._nodes['LayerNode.0'].ctrlWidget()
w.sizePolicy().setHorizontalPolicy(QtGui.QSizePolicy.MinimumExpanding)
self.tabWidget.currentChanged.connect(self.tabChanged)
self.tabWidget.setCurrentIndex(1)
self.tabWidget.setCurrentIndex(2)
self.fc.sigDisplayNodeAdded.connect(self.addPlot)
self.displayControlDock = pg.dockarea.Dock("Control")
formWidget = pg.QtGui.QWidget()
displayCtrlUi = DispCtrl.Ui_displayControlWidget()
displayCtrlUi.setupUi(formWidget)
displayCtrlUi.previousBtn.clicked.connect(self.updateDisplay)
displayCtrlUi.updateBtn.clicked.connect(self.updateDisplay)
displayCtrlUi.nextBtn.clicked.connect(self.updateDisplay)
self.displayCtrlUi = displayCtrlUi
self.displayControlDock.addWidget(formWidget)
self.ui.displayArea.addDock(self.displayControlDock, position="top")
self.solverNode = SolverNode("solver", solverFile, weightsFile, directory)
self.solverNode.sigOutputDataChanged.connect(self.updateTrainPlots)
self.ui.trainButton.pressed.connect(self.solverNode.trainNet)
self.ui.stopButton.pressed.connect(self.solverNode.stopTraining)
self.ui.resumeButton.pressed.connect(self.solverNode.resumeTraining)
self.ui.loadSolverButton.pressed.connect(self.loadSolver)
self.ui.saveSolverButton.pressed.connect(self.saveSolver)
self.ui.trainParamTree.addParameters(self.solverNode.param)
self.trainPlotItem = self.ui.trainGraphicsLayout.plotItem
def tabChanged(self, tabIndex):
"""handle moving the flowchart around"""
if tabIndex == 0:
self.setTab0()
elif tabIndex == 2:
self.setTab2()
# self.tabWidget.update()
def setTab0(self):
self.ui.flowchartCtrlWidget.layout().addWidget(self.fc.widget())
self.ui.flowchartWidget.layout().addWidget(self.fc.widget().chartWidget)
self.tabWidget.update()
def setTab2(self):
self.ui.flowchartCtrlWidget_2.layout().addWidget(self.fc.widget())
self.ui.flowchartWidget_2.layout().addWidget(self.fc.widget().chartWidget)
self.tabWidget.update()
def addPlot(self, plotNode):
"""
add an connect a plot
"""
name = plotNode.name()
plot = pg.PlotWidget(name=name)
self.plots[name] = plot
self.plotNodes[name] = plotNode
# get class of plot node to determine type of plot
if isinstance(plotNode, displayNodes):
plotNode.setPlot(plot)
plotDock = pg.dockarea.Dock(name)
self.plotDocks[name] = plotDock
plotDock.addWidget(plot)
self.ui.displayArea.addDock(plotDock)
self.tabWidget.update()
# for name, plotNode in self.plotNodes.items():
# plotNode.setPlotList(self.plots)
def updateDisplay(self):
sender = self.sender()
if sender.objectName() == u"previousBtn":
self.selectInput(-1)
elif sender.objectName() == u"nextBtn":
self.fc.process(forward=True)
else:
self.fc.process(forward=False)
def loadSolver(self, fileName=None, startDir=None):
if fileName is None:
if startDir is None:
startDir = self.directory
if startDir is None:
startDir = "."
self.fileDialog = FileDialog(None, "Load Flowchart..", startDir, "Protobuf (*.prototxt)")
self.fileDialog.show()
self.fileDialog.fileSelected.connect(self.loadSolver)
return
fileName = unicode(fileName)
self.solverNode.setProto(fileName)
def saveSolver(self, fileName=None, startDir=None, suggestedFileName="solver.prototxt"):
if fileName is None:
if startDir is None:
startDir = self.directory
if startDir is None:
startDir = "."
self.fileDialog = FileDialog(None, "Save Solver..", startDir, "Prototxt (*.prototxt)")
self.fileDialog.setAcceptMode(QtGui.QFileDialog.AcceptSave)
self.fileDialog.show()
self.fileDialog.fileSelected.connect(self.saveSolver)
return
fileName = unicode(fileName)
self.solverNode.writeProto(fileName)
def updateTrainPlots(self, data):
for lossName, color in dict(trainLoss="b", testLoss="g", testAcc="r").items():
for trainLossName, trainLossArr in data[lossName].items():
try:
self.trainLossPlots[trainLossName].setData(trainLossArr)
except KeyError:
self.trainLossPlots[trainLossName] = self.trainPlotItem.plot(
trainLossArr, pen=color, name=trainLossName
)
def stopTraining(self):
pass
def resumeTraining(self):
pass