diff --git a/fastdeploy/backends/ort/ort_backend.h b/fastdeploy/backends/ort/ort_backend.h index 10c841ab2..445d55540 100644 --- a/fastdeploy/backends/ort/ort_backend.h +++ b/fastdeploy/backends/ort/ort_backend.h @@ -66,14 +66,14 @@ class OrtBackend : public BaseBackend { const OrtBackendOption& option = OrtBackendOption(), bool from_memory_buffer = false); - bool Infer(std::vector& inputs, std::vector* outputs); + bool Infer(std::vector& inputs, std::vector* outputs) override; - int NumInputs() const { return inputs_desc_.size(); } + int NumInputs() const override { return inputs_desc_.size(); } - int NumOutputs() const { return outputs_desc_.size(); } + int NumOutputs() const override { return outputs_desc_.size(); } - TensorInfo GetInputInfo(int index); - TensorInfo GetOutputInfo(int index); + TensorInfo GetInputInfo(int index) override; + TensorInfo GetOutputInfo(int index) override; std::vector GetInputInfos() override; std::vector GetOutputInfos() override; static std::vector custom_operators_; diff --git a/fastdeploy/backends/paddle/paddle_backend.h b/fastdeploy/backends/paddle/paddle_backend.h index ce4b33d7c..43f5a4a17 100644 --- a/fastdeploy/backends/paddle/paddle_backend.h +++ b/fastdeploy/backends/paddle/paddle_backend.h @@ -67,14 +67,14 @@ class PaddleBackend : public BaseBackend { const std::string& model_file, const std::string& params_file, const PaddleBackendOption& option = PaddleBackendOption()); - bool Infer(std::vector& inputs, std::vector* outputs); + bool Infer(std::vector& inputs, std::vector* outputs) override; - int NumInputs() const { return inputs_desc_.size(); } + int NumInputs() const override { return inputs_desc_.size(); } - int NumOutputs() const { return outputs_desc_.size(); } + int NumOutputs() const override { return outputs_desc_.size(); } - TensorInfo GetInputInfo(int index); - TensorInfo GetOutputInfo(int index); + TensorInfo GetInputInfo(int index) override; + TensorInfo GetOutputInfo(int index) override; std::vector GetInputInfos() override; std::vector GetOutputInfos() override; diff --git a/fastdeploy/core/fd_type.cc b/fastdeploy/core/fd_type.cc index c31879b1d..850f38ca5 100644 --- a/fastdeploy/core/fd_type.cc +++ b/fastdeploy/core/fd_type.cc @@ -59,6 +59,22 @@ std::string Str(const Device& d) { return out; } +std::ostream& operator<<(std::ostream& out,const Device& d){ + switch (d) { + case Device::CPU: + out << "Device::CPU"; + break; + case Device::GPU: + out << "Device::GPU"; + break; + default: + out << "Device::UNKOWN"; + } + return out; +} + + + std::string Str(const FDDataType& fdt) { std::string out; switch (fdt) { @@ -95,6 +111,41 @@ std::string Str(const FDDataType& fdt) { return out; } +std::ostream& operator<<(std::ostream& out,const FDDataType& fdt){ + switch (fdt) { + case FDDataType::BOOL: + out << "FDDataType::BOOL"; + break; + case FDDataType::INT16: + out << "FDDataType::INT16"; + break; + case FDDataType::INT32: + out << "FDDataType::INT32"; + break; + case FDDataType::INT64: + out << "FDDataType::INT64"; + break; + case FDDataType::FP32: + out << "FDDataType::FP32"; + break; + case FDDataType::FP64: + out << "FDDataType::FP64"; + break; + case FDDataType::FP16: + out << "FDDataType::FP16"; + break; + case FDDataType::UINT8: + out << "FDDataType::UINT8"; + break; + case FDDataType::INT8: + out << "FDDataType::INT8"; + break; + default: + out << "FDDataType::UNKNOWN"; + } + return out; +} + template const FDDataType TypeToDataType::dtype = UNKNOWN1; diff --git a/fastdeploy/core/fd_type.h b/fastdeploy/core/fd_type.h index 22f33e0e5..f2746c51b 100644 --- a/fastdeploy/core/fd_type.h +++ b/fastdeploy/core/fd_type.h @@ -51,6 +51,10 @@ enum FASTDEPLOY_DECL FDDataType { INT8 }; +FASTDEPLOY_DECL std::ostream& operator<<(std::ostream& out,const Device& d); + +FASTDEPLOY_DECL std::ostream& operator<<(std::ostream& out,const FDDataType& fdt); + FASTDEPLOY_DECL std::string Str(const FDDataType& fdt); FASTDEPLOY_DECL int32_t FDDataTypeSize(const FDDataType& data_dtype);