caffeNetViewer.py 文件源码

def runCaffeModel(self):
        iname = str(self.ui.comboBoxImage.currentText())
        self.cImg = cv2.imread(iname)
        self.cImg = cv2.cvtColor(self.cImg, cv2.COLOR_BGR2RGB)
        self.ui.plainTextEdit.appendPlainText('Model Running ... ')
        self.ui.plainTextEdit.appendPlainText('  Image Name : '+iname)
        self.ui.plainTextEdit.appendPlainText("  Image Shape : " + str(self.cImg.shape))
        self.ui.plainTextEdit.appendPlainText("  Model Input Image Shape : " + str(self.net.blobs['data'].data.shape))  

        transformer = caffe.io.Transformer({'data': self.net.blobs['data'].data.shape})
        transformer.set_transpose('data', (2,0,1))  # move image channels to outermost dimension
        #transformer.set_mean('data', mu)            # subtract the dataset-mean value in each channel
        transformer.set_raw_scale('data', 255)      # rescale from [0, 1] to [0, 255]
        transformer.set_channel_swap('data', (2,1,0))  # swap channels from RGB to BGR          

        image = caffe.io.load_image(iname)
        inData = transformer.preprocess('data', image)

        self.net.blobs['data'].data[...] = [inData]

        self.outClass = self.net.forward()
        self.on_comboBoxLayers_currentIndexChanged()