You cannot select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
19 lines
615 B
C++
19 lines
615 B
C++
//
|
|
// Created by 庾金科 on 21/10/2017.
|
|
//
|
|
|
|
#include "CNNRecognizer.h"
|
|
|
|
namespace pr{
|
|
CNNRecognizer::CNNRecognizer(std::string prototxt,std::string caffemodel){
|
|
net = cv::dnn::readNetFromCaffe(prototxt, caffemodel);
|
|
}
|
|
|
|
label CNNRecognizer::recognizeCharacter(cv::Mat charImage){
|
|
if(charImage.channels()== 3)
|
|
cv::cvtColor(charImage,charImage,cv::COLOR_BGR2GRAY);
|
|
cv::Mat inputBlob = cv::dnn::blobFromImage(charImage, 1/255.0, cv::Size(CHAR_INPUT_W,CHAR_INPUT_H), cv::Scalar(0,0,0),false);
|
|
net.setInput(inputBlob,"data");
|
|
return net.forward();
|
|
}
|
|
} |