| 84 | } // namespace |
| 85 | |
| 86 | TfLiteStatus InputGenerator::LoadModel(const string& model_dir) { |
| 87 | model_ = FlatBufferModel::BuildFromFile(model_dir.c_str()); |
| 88 | if (!model_) { |
| 89 | fprintf(stderr, "Cannot load model %s", model_dir.c_str()); |
| 90 | return kTfLiteError; |
| 91 | } |
| 92 | |
| 93 | ::tflite::ops::builtin::BuiltinOpResolver builtin_ops; |
| 94 | InterpreterBuilder(*model_, builtin_ops)(&interpreter_); |
| 95 | if (!interpreter_) { |
| 96 | fprintf(stderr, "Failed to build interpreter."); |
| 97 | return kTfLiteError; |
| 98 | } |
| 99 | |
| 100 | return kTfLiteOk; |
| 101 | } |
| 102 | |
| 103 | TfLiteStatus InputGenerator::ReadInputsFromFile(const string& filename) { |
| 104 | if (filename.empty()) { |