| 164 | |
| 165 | #ifdef USE_OPENCV |
| 166 | cv::Mat DecodeDatumToCVMatNative(const Datum& datum) { |
| 167 | cv::Mat cv_img; |
| 168 | CHECK(datum.encoded()) << "Datum not encoded"; |
| 169 | const string& data = datum.data(); |
| 170 | std::vector<char> vec_data(data.c_str(), data.c_str() + data.size()); |
| 171 | cv_img = cv::imdecode(vec_data, -1); |
| 172 | if (!cv_img.data) { |
| 173 | LOG(ERROR) << "Could not decode datum "; |
| 174 | } |
| 175 | return cv_img; |
| 176 | } |
| 177 | cv::Mat DecodeDatumToCVMat(const Datum& datum, bool is_color) { |
| 178 | cv::Mat cv_img; |
| 179 | CHECK(datum.encoded()) << "Datum not encoded"; |