当前位置:网站首页>Teach yourself to train pytorch model to Caffe (III)
Teach yourself to train pytorch model to Caffe (III)
2022-07-05 21:09:00 【FeboReigns】
This section uses caffe To reason out pytorh The same result
Reference resources : Use Caffe Of Python Interface for reasoning - Simple books (jianshu.com)
#coding = utf-8
import numpy as np
import sys, os
sys.path.insert(0, caffe_root + 'python')
import caffe
os.chdir(caffe_root)
#caffe.set_device(0)
#caffe.set_mode_gpu()
#### The following three lines are modified according to their own path
net_file = './dock_googlenet_dog.prototxt'
caffe_model = './dock_googlenet_dog.caffemodel'
imagenet_labels_filename = './class_names.txt'
# load model
net = caffe.Net(net_file, caffe_model, caffe.TEST)
#resize
transformer = caffe.io.Transformer({'data': net.blobs['data'].data.shape})
# Get the number of channels to the front Turn into 【1,3 224,224】
transformer.set_transpose('data', (2, 0, 1))
# # Call the mean file
# # transformer.set_mean('data', np.load(mean_file).mean(1).mean(1))
#python Store pictures as [0, 1], and caffe Store pictures as [0, 255], So we need a transformation
# transformer.set_raw_scale('data', 255)
#caffe The picture is RGB Format , And network format BGR, To transform
# transformer.set_channel_swap('data', (2, 1, 0))
### Change according to your own path
im = caffe.io.load_image('./cat.2.jpg')
net.blobs['data'].data[...] = transformer.preprocess('data',im)
# net.blobs['data'].data[...] =
out = net.forward()
print (net.blobs['data'].data.shape)
print (net.blobs['data'].data.dtype)
print (net.blobs[net.blobs.keys()[-1]].data.shape)
print (net.blobs[net.blobs.keys()[-1]].data.dtype)
labels = np.loadtxt(imagenet_labels_filename, str, delimiter='\t')
top_k = net.blobs[net.blobs.keys()[-1]].data[0].flatten().argsort()[-1:-6:-1]
for i in np.arange(top_k.size):
print (top_k[i], labels[top_k[i]])
# Print the final score
print (net.blobs[net.blobs.keys()[-1]].data[0])
./class_names.txt Namely
0 dog
1 cat
Note that my training is not used normlize,
And my format is RGB Of , So I have a few useless preprocessing
caffe Of caffe.io.load_image('./cat.2.jpg') yes RGB Of
Mine is also. RBG Of
If you use brother's docker My code will report three errors when the environment runs
###got an unexpected keyword argument ‘as_grey‘
https://blog.csdn.net/hjxu2016/article/details/113541609
###TypeError: 'float' object cannot be interpreted as an integer
https://blog.csdn.net/qq_28634403/article/details/81224291
###dict_keys‘ object is not subscriptable
https://blog.csdn.net/le___le/article/details/103617431
I use my own configuration caffe, my caffe No, ceil mode Parameters , use round_mode Instead of the , If ceil mode by true be round_mode by 0, Otherwise, vice versa . Accuracy basically does not affect Pytorch maxpool Of ceil_mode And caffe in maxpool Comparison _zl3090 The blog of -CSDN Blog
How to match yourself caffe Look at my other blog :
finally ubuntu Finished compiling CPU edition caffe 了 _FeboReigns The blog of -CSDN Blog
边栏推荐
- php中explode函数存在的陷阱
- ArcGIS\QGIS无插件加载(无偏移)MapBox高清影像图
- PVC 塑料片BS 476-6 火焰传播性能测定
- PostGIS installation geographic information extension
- 显示器要申请BS 476-7 怎么送样?跟显示屏一样吗??
- 显示屏DIN 4102-1 Class B1防火测试要求
- Pytorch实战——MNIST数据集手写数字识别
- 水泥胶黏剂BS 476-4 不燃性测试
- The transformation based on vertx web sstore redis to realize the distributed session of vertx HTTP application
- 浅聊我和一些编程语言的缘分
猜你喜欢
随机推荐
第05章_存储引擎
Influence of oscilloscope probe on signal source impedance
获取前一天的js(时间戳转换)
Clickhouse copy paste multi line SQL statement error
XML建模
ts 之 属性的修饰符public、private、protect
驱动壳美国测试UL 2043 符合要求有哪些?
PostGIS installation geographic information extension
Utils/index TS tool function
SQL series (basic) - Chapter 2 limiting and sorting data
[case] Application of element display and hiding -- element mask
珍爱网微服务底层框架演进从开源组件封装到自研
秋招将临 如何准备算法面试、回答算法面试题
中国的软件公司为什么做不出产品?00后抛弃互联网;B站开源的高性能API网关组件|码农周刊VIP会员专属邮件周报 Vol.097
Dictionary tree simple introductory question (actually blue question?)
vant 源码解析之 utils/index.ts 工具函数
EasyExcel的讀寫操作
100 cases of shell programming
Aitm 2-0003 horizontal combustion test
Hdu2377bus pass (build more complex diagram +spfa)