优化原因:
原版的caffe ssd源码由于数据预处理速度很慢且数据预处理的速度远远跟不上前向和反向计算的速度,导致训练速度很慢,即训练遇到的性能瓶颈在数据预处理部分。
优化方法:
1、数据预处理对多幅图片的处理改为多线程;
2、数据预处理过程中,不需要将图片格式在datum和cv::Mat之间来回转换导致不必要的时间浪费;
优化后的源码请参考:
优化原因:
原版的caffe ssd源码由于数据预处理速度很慢且数据预处理的速度远远跟不上前向和反向计算的速度,导致训练速度很慢,即训练遇到的性能瓶颈在数据预处理部分。
优化方法:
1、数据预处理对多幅图片的处理改为多线程;
2、数据预处理过程中,不需要将图片格式在datum和cv::Mat之间来回转换导致不必要的时间浪费;
优化后的源码请参考: