#include "stdafx.h"
#include <opencv2/dnn.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/highgui.hpp>
using namespace cv;
using namespace cv::dnn;
#include <fstream>
#include <iostream>
#include <cstdlib>
using namespace std;
const size_t width = 256;
const size_t height = 256;
Mat getMean(const size_t& imageHeight, const size_t& imageWidth)
{
Mat mean;
const int meanValues[3] = { 104, 117, 123 };
vector<Mat> meanChannels;
for (size_t i = 0; i < 3; i++)
{
Mat channel(imageHeight, imageWidth, CV_32F, Scalar(meanValues[i]));
meanChannels.push_back(channel);
}
cv::merge(meanChannels, mean);
return mean;
}
Mat preprocess(const Mat& frame)
{
Mat preprocessed;
frame.convertTo(preprocessed, CV_32FC3);
resize(preprocessed, preprocessed, Size(width, height)); //SSD accepts 300x300 RGB-
images
Mat mean = getMean(width, height);
cv::subtract(preprocessed, mean, preprocessed);
return preprocessed;
}
const char* about = "This sample uses Single-Shot Detector"
"to detect objects on image"; // TODO: link
const char* params
= "{ help | false | print usage }"
"{ protot |F://caffee//mine//deploy.prototxt| model configuration }"
"{ model |F://caffee//mine//mo//_iter_2000.caffemodel| model weights }"
"{ image |F://shipin//qww.jpg| image for detection }"
"{ min_confidence | 0.5 | min confidence }";
资源评论
