Context Is King!分享 http://blog.sciencenet.cn/u/chauncey 计算机视觉

博文

caffe初测试(1)

已有 10461 次阅读 2014-8-8 17:21 |个人分类:Deep Learning|系统分类:科研笔记

1)   准备图片,并建立列表文件。

包括训练数据和验证数据。列表文件listfile结构为:filename label (subfolder/file1.jpeg 7)

2)   转换图片转为leveldb数据结构,并求图像平均

(1)  convert_imagesetrootfolder/ listfile DB_NAME [0/1]

rootfolder 为图像所在目录,listfile为图片以及标记的列表文件,最后一位选0表示安装给定顺序,1表示重新洗牌

注意:图像大小必须统一

(2)  compute_image_mean DB_NAME MEAN_NAME

3)   建立网络模型,并训练

主要是写train.prototxttest.prototxt,还有solver.prototxt。可以考虑修改imagenet的网络模型,需要修改:

   source: "**_leveldb"

   mean_file: "**_mean.binaryproto"


采用IMagenet设置,出现如下问题:

error == cudaSucess<9 vs. 0> invalid configuration argument

看来要修改参数了,搜了一下,应该是batch_size吧。

等待结果。。。




https://blog.sciencenet.cn/blog-43092-818118.html

上一篇:caffe 资料收集
下一篇:如何在编号数字前添加上自定义的字母
收藏 IP: 202.114.10.*| 热度|

0

该博文允许注册用户评论 请点击登录 评论 (0 个评论)

数据加载中...

Archiver|手机版|科学网 ( 京ICP备07017567号-12 )

GMT+8, 2024-12-23 21:23

Powered by ScienceNet.cn

Copyright © 2007- 中国科学报社

返回顶部