Fork me on GitHub

每次运行caffe代码之前需要考虑修改的地方

   

Train阶段:

(1)       train.sh中的solver.prototxt路径

(2)       train.sh中的caffemodel路径

(3)       train.sh中的日志名称

(4)       solver.prototxt中的net名称(也就是train.prototxt路径)

(5)       solver.prototxt中的snapshot_prefix名称(包括路径+文件名)

(6)       train.prototxt中的data层(有两个,TRAIN和TEST)输入文件的路径,即list.txt或者lmdb的路径

Test阶段:

(1)       test.py脚本中的caffemodel改成训练好的caffemodel

(2)       test.py中的prototxt改成deploy.prototxt

(3)       test.py中的labelfile(即list.txt文件的路径)

posted @ 2019-02-23 08:59  stardsd  阅读(317)  评论(0编辑  收藏  举报