opencv/opencv_contrib
1
0
Fork 0
mirror of https://github.com/opencv/opencv_contrib.git synced 2025-10-19 02:16:34 +08:00
Code Issues Releases Wiki Activity

Fixing warnings

Browse Source
This commit is contained in:
Anguelos Nicolaou
2016-07-07 15:42:39 +02:00
parent 551d1e684c
commit 7c7ecd71d2
1 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
modules/text/src/ocr_holistic.cpp
View File

@@ -60,9 +60,9 @@ protected:
void classifyMiniBatch(std::vector<Mat> inputImageList, Mat outputMat){
//Classifies a list of images containing at most minibatchSz_ images
CV_Assert(inputImageList.size()<=this->minibatchSz_);
CV_Assert(int(inputImageList.size())<=this->minibatchSz_);
CV_Assert(outputMat.isContinuous());
float* ouputPtr= (float*)(outputMat.data);
#ifdef HAVE_CAFFE
net_->input_blobs()[0]->Reshape(inputImageList.size(), 1,this->inputGeometry_.height,this->inputGeometry_.width);
net_->Reshape();
float* inputBuffer=net_->input_blobs()[0]->mutable_cpu_data();
@@ -78,6 +78,7 @@ protected:
const float* outputNetData=net_->output_blobs()[0]->cpu_data();
float*outputMatData=(float*)(outputMat.data);
memcpy(outputMatData,outputNetData,sizeof(float)*this->outputSize_*inputImageList.size());
#endif
}
#ifdef HAVE_CAFFE
@@ -129,7 +130,7 @@ public:
inputImageList.getMatVector(allImageVector);
classProbabilities.create(Size(this->outputSize_,allImageVector.size()),CV_32F);
Mat outputMat = classProbabilities.getMat();
for(int imgNum=0;imgNum<allImageVector.size();imgNum+=this->minibatchSz_){
for(int imgNum=0;imgNum<int(allImageVector.size());imgNum+=this->minibatchSz_){
int rangeEnd=imgNum+std::min<int>(allImageVector.size()-imgNum,this->minibatchSz_);
std::vector<Mat>::const_iterator from=allImageVector.begin()+imgNum;
std::vector<Mat>::const_iterator to=allImageVector.begin()+rangeEnd;
@@ -208,7 +209,7 @@ public:
while (std::getline(labelsFile, line)){
labels_.push_back(std::string(line));
}
CV_Assert(this->classifier_->getOutputSize()==this->labels_.size());
CV_Assert(this->classifier_->getOutputSize()==int(this->labels_.size()));
}
void recogniseImage(InputArray inputImage,CV_OUT String& transcription,CV_OUT double& confidence){