1 #ifndef HAAS_ML_OLDA_H 2 #define HAAS_ML_OLDA_H 3 #include "base/include/IHaasML.h" 4 5 class HaasMLOlda : public IHaasML 6 { 7 public: 8 HaasMLOlda(); 9 virtual ~HaasMLOlda(); 10 11 virtual int SetInputData(const char* dataPath); 12 virtual int LoadNet(const char* modePath); 13 virtual int Predict(); 14 virtual int GetPredictResponses(char* outResult, int len); 15 virtual int UnLoadNet(); 16 private: 17 int uai_get_img_dec_result(int8_t output_softmax[], int result_num, char *result); 18 uint32_t input_scale; 19 uint32_t input_shift; 20 uint8_t *norm_img_data; 21 int8_t out_output_Softmax[21]; 22 }; 23 #endif 24