DataRow.py 文件源码

def predict(self, resized):
        """
        @resized: image 40,40 already pre processed 
        """         
        #self.net.blobs['data'].data[...] = cv2.split(resized)
        self.net.blobs['data'].data[...] = resized.reshape(1,1,60,60)
        prediction = self.net.forward()['Dense3'][0]
        return prediction