caffeNetViewer.py 文件源码

def loadLayers(self):
        mname = str(self.ui.comboBoxModel.currentText())
        wname = str(self.ui.comboBoxWeights.currentText())
        self.net = caffe.Net(mname,wname , caffe.TEST)
        out = self.net.blobs
        self.layerList = out.keys()
        self.ui.comboBoxLayers.clear()
        for ln in self.layerList :
            self.ui.comboBoxLayers.addItem(ln)

        self.ui.plainTextEdit.clear()
        self.ui.plainTextEdit.appendPlainText('Caffe Model Loaded...')
        self.ui.plainTextEdit.appendPlainText('  Model Name : '+mname)
        self.ui.plainTextEdit.appendPlainText('  Weights Name : '+wname)

        self.ui.plainTextEdit.appendPlainText("Network Layers ...")

        for name, layer in zip(self.net._layer_names, self.net.layers):
            if not name in self.layerList :
                continue
            msg = "   "+name +" --> "+str(layer.type) +" --> "+ str((self.net.blobs[name].data[0]).shape)
            self.ui.plainTextEdit.appendPlainText(msg)      

        self.isModelLoaded = True