Commit 8475c0bb authored by Henry's avatar Henry Committed by Yongzhe Wang
Browse files

Fix segmentation fault on SSD tflite client destruction. (#7804)

* Fix segmentation fault on client destruction.

edgetpu::EdgeTpuContext needs to be created before tflite::Interpreter
parent 8861a624
...@@ -72,14 +72,14 @@ class MobileSSDTfLiteClient : public MobileSSDClient { ...@@ -72,14 +72,14 @@ class MobileSSDTfLiteClient : public MobileSSDClient {
virtual bool IsQuantizedModel() const; virtual bool IsQuantizedModel() const;
std::unique_ptr<::tflite::FlatBufferModel> model_;
std::unique_ptr<::tflite::MutableOpResolver> resolver_;
std::unique_ptr<::tflite::Interpreter> interpreter_;
#ifdef ENABLE_EDGETPU #ifdef ENABLE_EDGETPU
std::unique_ptr<edgetpu::EdgeTpuContext> edge_tpu_context_; std::unique_ptr<edgetpu::EdgeTpuContext> edge_tpu_context_;
#endif #endif
std::unique_ptr<::tflite::FlatBufferModel> model_;
std::unique_ptr<::tflite::MutableOpResolver> resolver_;
std::unique_ptr<::tflite::Interpreter> interpreter_;
private: private:
// MobileSSDTfLiteClient is neither copyable nor movable. // MobileSSDTfLiteClient is neither copyable nor movable.
MobileSSDTfLiteClient(const MobileSSDTfLiteClient&) = delete; MobileSSDTfLiteClient(const MobileSSDTfLiteClient&) = delete;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment