| 102 | } |
| 103 | |
| 104 | CaffePreproccessor::CaffePreproccessor(std::array<float, 3> mean, bool bgr, float scale) |
| 105 | : _mean(mean), _bgr(bgr), _scale(scale) |
| 106 | { |
| 107 | if (_bgr) |
| 108 | { |
| 109 | std::swap(_mean[0], _mean[2]); |
| 110 | } |
| 111 | } |
| 112 | |
| 113 | void CaffePreproccessor::preprocess(ITensor &tensor) |
| 114 | { |