#include"run.h"
|
|
bool RUN::initConfig(const char* modelpath, int* beltregion, float threshold) {
|
net = cv::dnn::readNetFromONNX(modelpath);
|
x1 = beltregion[0];
|
y1 = beltregion[1];
|
x2 = beltregion[2];
|
y2 = beltregion[3];
|
threshold_ = threshold;
|
|
return true;
|
}
|
|
int RUN::detect(std::vector<cv::Mat>& images)
|
{
|
try
|
{
|
cv::Rect roi(x1, y1, x2 - x1, y2 - y1);
|
std::vector<cv::Mat> r_ims;
|
for (int i = 0; i < 3; ++i) {
|
std::vector<cv::Mat> splited;
|
cv::Mat resized;
|
cv::split(images[i](roi), splited);
|
cv::resize(splited[2], resized, cv::Size(224, 224));
|
r_ims.emplace_back(resized); }
|
|
cv::Mat input_image(cv::Size(224, 224), CV_32FC3);
|
merge(r_ims, input_image);
|
cv::Mat blob = cv::dnn::blobFromImage(input_image, 1 / 255.);
|
net.setInput(blob);
|
std::vector<cv::Mat> outputs;
|
net.forward(outputs, net.getUnconnectedOutLayersNames());
|
float* data = (float*)outputs[0].data;
|
|
|
if (data[0] > threshold_)
|
{
|
return 0;//运行
|
}
|
else
|
{
|
return 1;//停止
|
}
|
}
|
catch (const std::exception& ex)
|
{
|
std::string errorMessage = "runModel-";
|
errorMessage += ex.what();
|
throw std::runtime_error(errorMessage);
|
}
|
|
}
|
