Skip to content

jren2017/imcl-caffe

 
 

Repository files navigation

#caffe环境搭建 1.安装CUDA7.5

2.把3rdparty解压到根目录

3.双击./src/caffe/proto/extract_proto.bat批处理文件来生成caffe.pb.h和caffe.pb.cc两个c++文件,和caffe_pb2.py这个Python使用的文件

4.打开工程文件buildVS2013/MainBuilder.sln,重新生成所有工程

中文博客

#python环境搭建 1.安装Anaconda

2.安装pycharm

#imcl-caffe使用说明 ##生成数据 根目录下python文件夹里面包含了实验所用到的数据生成等脚本

主要文件使用如下:

gen_test_train.py:将总数据划分为训练集和测试集

label_file_util.py:读取caffe格式或者文本格式图像标签文本文件

genh5_data.py:生成实验数据.h5文件

predict3.py:实验测试 ##开始训练 根目录下

run_surtest.bat:用实验室Alex_net网络训练,需要在自己环境下修改相应目录路径

snapshot.bat:在上一次断点处继续训练

weights.bat:调整训练权重再训练

About

windows环境caffe深度学习框架

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • C++ 60.2%
  • Jupyter Notebook 18.1%
  • Python 7.7%
  • Protocol Buffer 7.5%
  • Cuda 5.2%
  • MATLAB 0.9%
  • Other 0.4%