博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
在caffe中用训练好的 caffemodel 来分类新的图片所遇到的问题
阅读量:4603 次
发布时间:2019-06-09

本文共 315 字,大约阅读时间需要 1 分钟。

结合之前的博客: http://www.cnblogs.com/Allen-rg/p/5834551.html#3949333

用caffemodel去测试单通道的图像(mnist数据集)时,出现了问题,例如:维度不匹配、单通道图像和三通道图像不能强制转换等问题。

因为链接中的代码适用于对RGB三通道的图像的分类。

只需要将代码中: im=caffe.io.load_image(img)  #加载图片 

修改为: im=caffe.io.load_image(img,False) 即可

 

将图像读入修改为单通道的图像读入即可;

转载于:https://www.cnblogs.com/Allen-rg/p/8805534.html

你可能感兴趣的文章
存储过程Oracle(一)
查看>>
log4j日志归档
查看>>
Java笔记01——IO流
查看>>
mysql遇见error,1049
查看>>
NYOJ311 完全背包
查看>>
codevs——2822 爱在心中
查看>>
Python基础班---第一部分(基础)---Python基础知识---认识Python
查看>>
JAVA MAC 配置
查看>>
1134 最长上升子序列 (序列型 DP)
查看>>
js冒泡排序
查看>>
第一次作业 4班卢炳武
查看>>
抽象类的调用
查看>>
使用硬盘,安装双系统,Win7+CentOS
查看>>
Javascript学习总结
查看>>
php 用正则替换中文字符一系列问题解决
查看>>
ActiveMQ应用笔记一:基本概念&安装
查看>>
大话数据结构之四(串)
查看>>
加热炉简是新来的整个系统的板
查看>>
Mockito使用注意事项
查看>>
[LeetCode] Palindrome Linked List 回文链表
查看>>