137 lines
4.7 KiB
C++
137 lines
4.7 KiB
C++
// XuAilibTest.cpp : 此文件包含 "main" 函数。程序执行将在此处开始并结束。
|
||
//
|
||
//#pragma comment(lib,LIBPATH(__FILE__, "..\\srilm-lib\\common.lib"))
|
||
#include <iostream>
|
||
#include <SegDetector.h>
|
||
#include <NanoDetector.h>
|
||
#include <opencv2\imgproc\types_c.h>
|
||
#include<opencv2/imgproc/imgproc_c.h>
|
||
#include "LineHandle.h"
|
||
#include<numeric>
|
||
#include<windows.h>
|
||
#include <DoublePointerAlgorithm.h>
|
||
#include "SemicircularMeter.h"
|
||
#include <BrowseForFolder.h>
|
||
using namespace std;
|
||
using namespace cv;
|
||
|
||
int main()
|
||
{
|
||
|
||
const char* pstrModelFile = "Model_det.m";
|
||
CNanoDetector cNanoDetector(pstrModelFile);
|
||
cNanoDetector.Init();
|
||
//LPCWSTR dir = TEXT("C:\\Users\\77411\\Desktop\\Images4\\");
|
||
|
||
//ShellExecute(NULL, NULL, dir, NULL, NULL, SW_SHOWNORMAL);
|
||
string path = "";
|
||
BrowseForFolder browseForFolder;
|
||
browseForFolder.SelectFile(path);
|
||
|
||
Mat mat = imread(path);
|
||
if ( mat.data!=NULL)
|
||
{
|
||
cNanoDetector.Execute(mat);
|
||
vector<DetectorResult> vec;
|
||
cNanoDetector.GetResults(vec);
|
||
if (vec.size()>0)
|
||
{
|
||
for (size_t i = 0; i < vec.size(); i++)
|
||
{
|
||
int width = vec[i].width;
|
||
int height = vec[i].height;
|
||
Rect m_select = Rect(vec[i].left, vec[i].top, width, height);
|
||
if ( string("biaopan").compare( vec[i].name ) ==0 )
|
||
{
|
||
Mat roi = mat(m_select);
|
||
DoublePointerAlgorithm doublePointerAlgorithm;
|
||
int result = doublePointerAlgorithm.GetResult(roi, true);
|
||
cout << "biaopan结果 ：" << result << endl;
|
||
}
|
||
if (string("ceshi").compare(vec[i].name) == 0)
|
||
{
|
||
Mat roi = mat(m_select);
|
||
SemicircularMeter semicircularMeter;
|
||
double result = semicircularMeter.GetResult(roi, false);
|
||
cout << "ceshi结果 ：" << result << endl;
|
||
}
|
||
}
|
||
waitKey(0);
|
||
//// destroyAllWindows();
|
||
}
|
||
}
|
||
string str;
|
||
std::cin >> str;
|
||
}
|
||
|
||
|
||
|
||
void CVmatToDatumEx(const std::vector<cv::Mat> images, Datum& datum, float input_scale = 1.f)
|
||
{
|
||
Shape shape(images.size(), images[0].channels(), images[0].rows, images[0].cols);
|
||
datum.Reshape(shape);
|
||
cv::Size input_size = images[0].size();
|
||
const int channels = images[0].channels();
|
||
datum.SetDataItemSize(1);
|
||
uchar* datum_data = (uchar*)datum.Getdata();
|
||
|
||
unsigned int offsetN = shape.C * shape.H * shape.W;
|
||
|
||
for (int n = 0; n < images.size(); n++)
|
||
{
|
||
const cv::Mat& image = images[n];
|
||
assert(input_size == image.size());
|
||
assert(channels == image.channels());
|
||
|
||
if (image.isContinuous())
|
||
{
|
||
memcpy(datum_data + n * offsetN, image.data, offsetN);
|
||
}
|
||
}
|
||
|
||
}
|
||
|
||
void CVmatVecToDatum(const std::vector<cv::Mat> images, Datum& datum, float input_scale = 1.f)
|
||
{
|
||
return CVmatToDatumEx(images, datum);
|
||
//all input image shape must be same.
|
||
Shape shape(images.size(), images[0].channels(), images[0].rows, images[0].cols);
|
||
datum.Reshape(shape);
|
||
cv::Size input_size = images[0].size();
|
||
const int channels = images[0].channels();
|
||
float* datum_data = (float*)datum.Getdata();
|
||
|
||
for (int n = 0; n < images.size(); n++)
|
||
{
|
||
const cv::Mat& image = images[n];
|
||
assert(input_size == image.size());
|
||
assert(channels == image.channels());
|
||
|
||
unsigned int offsetN = n * shape.C * shape.H * shape.W;
|
||
for (int h = 0; h < image.rows; h++)
|
||
{
|
||
const uchar* ptr = image.ptr<uchar>(h);
|
||
int img_index = 0;
|
||
for (int w = 0; w < image.cols; ++w)
|
||
{
|
||
for (int c = 0; c < image.channels(); ++c)
|
||
{
|
||
int datum_index = (c * image.rows + h) * image.cols + w + offsetN;
|
||
float value = (static_cast<float>(ptr[img_index++]));
|
||
datum_data[datum_index] = value * (input_scale / 255.f);;
|
||
}
|
||
}
|
||
}
|
||
}
|
||
}
|
||
// 运行程序: Ctrl + F5 或调试 >“开始执行(不调试)”菜单
|
||
// 调试程序: F5 或调试 >“开始调试”菜单
|
||
|
||
// 入门使用技巧:
|
||
// 1. 使用解决方案资源管理器窗口添加/管理文件
|
||
// 2. 使用团队资源管理器窗口连接到源代码管理
|
||
// 3. 使用输出窗口查看生成输出和其他消息
|
||
// 4. 使用错误列表窗口查看错误
|
||
// 5. 转到“项目”>“添加新项”以创建新的代码文件，或转到“项目”>“添加现有项”以将现有代码文件添加到项目
|
||
// 6. 将来，若要再次打开此项目，请转到“文件”>“打开”>“项目”并选择 .sln 文件
|