ImageRecognition/CircularArresterCurrentALGO.h

#pragma once
#define _CRT_SECURE_NO_WARNINGS
#include <iostream>
#include <fstream>
#include <string>
#include <math.h>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <onnxruntime_cxx_api.h>

using namespace cv;
using namespace std;
using namespace Ort;


class CircularArresterCurrentALGO
{
public:
	void Init();
	float detect(Mat& cv_image);
	//Mat creat_line_image(const Mat& circle, int Radius, int RingStride);
	//float get_meter_reader(const Mat& image);

private:
	int inpWidth;
	int inpHeight;
	int outWidth;
	int outHeight;

	const float pi = 3.1415926536f;
	const int circle_center[2] = { 256, 256 };
	const int CV_AA = 16;
	const float mean[3] = { 0, 0, 0 }, stds[3]= { 0.6f, 0.6f, 0.6f };

	Env env = Env(ORT_LOGGING_LEVEL_ERROR, "u2net");
	Ort::Session* ort_session = nullptr;
	SessionOptions sessionOptions = SessionOptions();
	vector<char*> input_names;
	vector<char*> output_names;
	vector<vector<int64_t>> input_node_dims; // >=1 outputs
	vector<vector<int64_t>> output_node_dims; // >=1 outputs
};