caffe ssd 优化

一、数据预处理加速(提升 4~6 倍) 优化原因: 原版 caffe ssd 数据预处理速度很慢(尤其是遇到 depthwise conv 的时候) 数据预处理的速度远远跟不上前向和反向计算的速度,导致训练速度很慢 优化方法: 数据预处理对多幅图片的处理改为多线程 数据预处理过程中,不需要将图片格式在 datum 和 cv::Mat 之间来回转换导致不必要的时间浪费 参考代码 https://github.com/maidabu/caffe-ssd-optimized 替换 caffe ssd 源代码中的相应层 二、多卡加速:NVIDIA NCCL 优化原因: 使用多卡加载预训练模型进行训练