caffe2
武小胖儿
这个作者很懒,什么都没留下…
展开
-
code and note for creating your own NN by caffe2
""" Created on Sun Apr 28 10:00:14 2019 @author: wujing """ from caffe2.python import workspace, model_helper import numpy as np # ===================================================================...原创 2019-04-28 12:06:32 · 368 阅读 · 0 评论 -
获取模型执行过程中的中间结果
def testImage(self,image_location,image_size): mean = self.readMeanFile() image = ImageProcess(image_location,image_size,mean).getImage() fileOp = FileOperation() #1.read t...原创 2019-05-03 13:46:49 · 94 阅读 · 0 评论 -
调用caffe2中pre-trained model(squeezenet)
原文链接:https://github.com/caffe2/tutorials/blob/master/Loading_Pretrained_Models.ipynb model源码:https://github.com/caffe2/models 注释:本文会修改原文的部分内容。 #!/usr/bin/env python2 from caffe2.python import works...原创 2019-04-29 12:26:15 · 487 阅读 · 0 评论 -
切分init_net
备注:原始的init加密过,不能直接使用,因此只能从源头寻找weights。幸运的是所有init_net和predict_net都采用了一种名叫proto的组织方式,所有我们学习一下proto. ''' procedure 1. input the binary data and use the X_pd2.py to parse the binary data 2. save the stru...原创 2019-04-30 16:40:31 · 154 阅读 · 0 评论