ImageRecognition/CircularOilSinglePointALGO.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>
#include <string>
#include <math.h>
#include <cmath>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/highgui/highgui.hpp>
#include "opencv2/imgproc/types_c.h"
#include <onnxruntime_cxx_api.h>
#include <vector>

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


class CircularOilSinglePointALGO
{
private:
	int inpWidth;
	int inpHeight;
	int outWidth;
	int outHeight;
	const float mean[3] = { 0.5f,0.5f,0.5f };
	const float stds[3] = { 0.5f,0.5f,0.5f };
	const float pi = 3.1415926536f;
	const int circle_center[2] = { 256, 256 };//300,238  300,255  
	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
public:
	void Init(bool isCuda);
	float detect(Mat& cv_image);
	Mat creat_line_image(const Mat& circle, int Radius, int RingStride);
	float get_meter_reader(const Mat& image);


};