Newer
Older
BleedingDetectionKimura-sanMethod / slice_predict.h
Lei Yang Lei Yang on 28 Feb 2022 753 bytes Add files via upload
#pragma once
#include "raw_image_class.h"
#include <vector>
//#include <myOpenCV.h>
//#include <mymist_ver2.h>
#include "myOpenCV3.h"
#include "mymist.h"
//using namespace cv;

//MIP画像からの肩及び腰のスライス位置推定に関するクラス
//DNNモジュールを用いてkerasで学習したモデルを読み込み推定(.pbファイルへの変換が必要)
static class Slice_predict
{
	//MIP画像生成
	static cv::Mat createMIP(Raw_image<short>& images);

	//相関係数の計算
	static std::vector<float> Correlation(std::vector<float>& input1, std::vector<float>& input2);

public:
	//スライス位置推定
	static void predict(Raw_image<short>& image, int& shoulder, int& low,
		cv::dnn::Net& class_model, cv::dnn::Net& shoulde_model,
		cv::dnn::Net& lower_model);
		
};