caffe训练检测过程注意事项

  1. 标签问题
    在制作VOC数据集的时候,注意标签不能弄错!!!!
  2. 图片大小问题:
    图片如果太大,就可以根据Mobilenet-ssd下的 slover_prototxt文件修改图片大小,,具体方法为 修改create_data.sh文件 里的 width 和 height参数 ,默认为0。若是修改之后 ,训练样本的时候速度明显变快。

3.训练的时候 LMDB数据文件的路径一定要给对 ,否则会报 check failed:mdb——status ==(2 VS 0)错误。

  1. 如使用python 调用caffe gpu 进行目标检测 ,可以在python文件中加上一行:caffe.set_mode_gpu() 即可调用GPU进行目标检测 。

  2. caffe训练的时候 可以在 train.sh文件总设置 GPU模式进行训练 - gpu 0,1,2,3 数字代表自己电脑的GPU编号 如使用cpu进行训练 可注释掉本行 然后再solver_train_prototxt文件中 修改solver_mode 为CPU。

  3. 训练的时候可把solver_train_prototxt文件中的base_lr 学习率调高一点 :0.001

  4. caffe支持snapshot训练,在训练的途中可以加入数据集数据进行 训练,但是数据的类不能改变。 如训练的时候停止了可以在 train.sh文件中修改 中的

-weight="mobilenet_iter_73000.caffemodel"
to 
-snapshot="snapshot/mobilenet_iter_20000.solverstate" //步数是你上次停止之前snapshot中最后保存的步数

8.注意修改batch_size的大小 以防内存不够 训练报 CUDA out of memory 的错误

©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

相关阅读更多精彩内容

友情链接更多精彩内容