如何使 Torch Caffe Lua中的Flickr Style示例工作?
2015-5-30 10:13:27
收藏:0
阅读:78
评论:0
使用[Torch Caffe Binding](https://github.com/szagoruyko/torch-caffe-binding) 。我想在[Flickr样式示例](http://caffe.berkeleyvision.org/gathered/examples/finetune_flickr_style.html)上进行预测。我已经拥有了训练好的模型和下面的代码。我如何更改代码才能使其工作?
require 'caffe'
require 'image'
-- 测试 Flickr 样式单个图像
-- 配置
caffe_path = '/home/user/caffe/'
data_path = caffe_path..'data/flickr_style/images/'
file_name = data_path..'614893901_c9ed490f80.jpg'
-- Caffe 运行在 CPU 模式
caffe.Net.setModeCPU()
-- 初始化网络
net = caffe.Net(caffe_path..'models/finetune_flickr_style/deploy.prototxt', caffe_path..'models/finetune_flickr_style/finetune_flickr_style.caffemodel','test')
-- 加载并调整图像大小
input = torch.FloatTensor(10,3,227,227)
img = image.load(file_name,3,'float')
img = image.scale(img,227,227)
for i=1,3 do
for j=1,32 do
for k=1,32 do
input[1][i][j][k] = img[i][j][k]
end
end
end
-- 预测结果
output = net:forward(input)
print(output[1])
y,j = torch.max(output[1],1)
print('Class: ', j[1][1][1] , y[1][1][1])
现在输出的是一个FloatTensor [20x1x1],其中有20个NaN值。
点赞
评论区的留言会收到邮件通知哦~
推荐文章
- 如何将两个不同的lua文件合成一个 东西有点长 大佬请耐心看完 我是小白研究几天了都没搞定
- 如何在roblox studio中1:1导入真实世界的地形?
- 求解,lua_resume的第二次调用继续执行协程问题。
- 【上海普陀区】内向猫网络招募【Skynet游戏框架Lua后端程序员】
- SF爱好求教:如何用lua实现游戏内调用数据库函数实现账号密码注册?
- Lua实现网站后台开发
- LUA错误显式返回,社区常见的规约是怎么样的
- lua5.3下载库失败
- 请问如何实现文本框内容和某个网页搜索框内容连接,并把网页输出来的结果反馈到另外一个文本框上
- lua lanes多线程使用
- 一个kv数据库
- openresty 有没有比较轻量的 docker 镜像
- 想问一下,有大佬用过luacurl吗
- 在Lua执行过程中使用Load函数出现问题
- 为什么 neovim 里没有显示一些特殊字符?
- Lua比较两个表的值(不考虑键的顺序)
- 有个lua简单的项目,外包,有意者加微信 liuheng600456详谈,最好在成都
- 如何在 Visual Studio 2022 中运行 Lua 代码?
- addEventListener 返回 nil Lua
- Lua中获取用户配置主目录的跨平台方法