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

Posted 2021-02-10 sddai

   

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文件的路径）