| 143 | #endif // USE_OPENCV |
| 144 | |
| 145 | bool ReadFileToDatum(const string& filename, const int label, |
| 146 | Datum* datum) { |
| 147 | std::streampos size; |
| 148 | |
| 149 | fstream file(filename.c_str(), ios::in|ios::binary|ios::ate); |
| 150 | if (file.is_open()) { |
| 151 | size = file.tellg(); |
| 152 | std::string buffer(size, ' '); |
| 153 | file.seekg(0, ios::beg); |
| 154 | file.read(&buffer[0], size); |
| 155 | file.close(); |
| 156 | datum->set_data(buffer); |
| 157 | datum->set_label(label); |
| 158 | datum->set_encoded(true); |
| 159 | return true; |
| 160 | } else { |
| 161 | return false; |
| 162 | } |
| 163 | } |
| 164 | |
| 165 | #ifdef USE_OPENCV |
| 166 | cv::Mat DecodeDatumToCVMatNative(const Datum& datum) { |