Skip to content

Commit 887e6e5

Browse files
committed
removed hard codinginput parameters
1 parent efc864c commit 887e6e5

File tree

2 files changed

+15
-15
lines changed

2 files changed

+15
-15
lines changed

modules/text/src/ocr_holistic.cpp

Lines changed: 7 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -271,9 +271,9 @@ class DeepCNNOpenCvDNNImpl: public DeepCNN{
271271
//Size outputGeometry_;//= Size(1,1);
272272
//int channelCount_;
273273
// int inputChannel_ ;//=1;
274-
//const int _inputHeight =32;
275-
//const int _inputWidth =100;
276-
//const int _inputChannel =1;
274+
int _inputHeight;
275+
int _inputWidth ;
276+
int _inputChannel ;
277277
public:
278278
DeepCNNOpenCvDNNImpl(const DeepCNNOpenCvDNNImpl& dn):
279279
minibatchSz_(dn.minibatchSz_),outputSize_(dn.outputSize_){
@@ -300,8 +300,8 @@ class DeepCNNOpenCvDNNImpl: public DeepCNN{
300300
//Implemented to supress Visual Studio warning "assignment operator could not be generated"
301301
}
302302

303-
DeepCNNOpenCvDNNImpl(String modelArchFilename, String modelWeightsFilename,Ptr<ImagePreprocessor> preprocessor, int maxMinibatchSz,int inputWidth =100,int inputHeight = 32)
304-
:minibatchSz_(maxMinibatchSz),_inputWidth(inputWidth),_inputHeight(inputHeight)
303+
DeepCNNOpenCvDNNImpl(String modelArchFilename, String modelWeightsFilename,Ptr<ImagePreprocessor> preprocessor, int maxMinibatchSz,int inputWidth =100,int inputHeight = 32,int inputChannel =1)
304+
:minibatchSz_(maxMinibatchSz),_inputWidth(inputWidth),_inputHeight(inputHeight),_inputChannel(inputChannel)
305305
{
306306

307307
CV_Assert(this->minibatchSz_>0);
@@ -612,13 +612,13 @@ class OCRHolisticWordRecognizerImpl: public OCRHolisticWordRecognizer{
612612
this->classifier_->classifyBatch(inputImageList,netOutput);
613613

614614
for(int k=0;k<netOutput.rows;k++)
615-
{https://www.google.es/?gfe_rd=cr&dcr=0&ei=4fq7We8Bk9jyB8zPp5AL
615+
{
616616
int classNum;
617617
double confidence;
618618
NetOutput::getClassification((float*)(netOutput.row(k).data),this->classifier_->getOutputSize(),classNum,confidence);
619619
transcriptionVec.push_back(this->labels_[classNum]);
620620
confidenceVec.push_back(confidence);
621-
}https://www.google.es/?gfe_rd=cr&dcr=0&ei=4fq7We8Bk9jyB8zPp5AL
621+
}
622622
}
623623

624624

modules/text/src/text_detectorCNN.cpp

Lines changed: 8 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -252,9 +252,9 @@ class DeepCNNTextDetectorDNNImpl: public DeepCNNTextDetector{
252252
//Size inputGeometry_;
253253
int minibatchSz_;//The existence of the assignment operator mandates this to be nonconst
254254
//int outputSize_;
255-
const int _inputHeight =700;
256-
const int _inputWidth =700;
257-
const int _inputChannel =3;
255+
int inputHeight_;
256+
int inputWidth_;
257+
int inputChannel_;
258258
public:
259259
DeepCNNTextDetectorDNNImpl(const DeepCNNTextDetectorDNNImpl& dn):
260260
minibatchSz_(dn.minibatchSz_){
@@ -282,8 +282,8 @@ class DeepCNNTextDetectorDNNImpl: public DeepCNNTextDetector{
282282
//Implemented to supress Visual Studio warning "assignment operator could not be generated"
283283
}
284284

285-
DeepCNNTextDetectorDNNImpl(String modelArchFilename, String modelWeightsFilename,Ptr<ImagePreprocessor> preprocessor, int maxMinibatchSz)
286-
:minibatchSz_(maxMinibatchSz)
285+
DeepCNNTextDetectorDNNImpl(String modelArchFilename, String modelWeightsFilename,Ptr<ImagePreprocessor> preprocessor, int maxMinibatchSz,int inputHeight=700,int inputWidth =700,int inputChannel =3)
286+
:minibatchSz_(maxMinibatchSz),inputHeight_(inputHeight),inputWidth_(inputWidth),inputChannel_(inputChannel)
287287
{
288288

289289
CV_Assert(this->minibatchSz_>0);
@@ -304,9 +304,9 @@ class DeepCNNTextDetectorDNNImpl: public DeepCNNTextDetector{
304304
exit(-1);
305305
}
306306

307-
this->inputGeometry_.height =_inputHeight;
308-
this->inputGeometry_.width = _inputWidth ;//inputLayer->width();
309-
this->inputChannelCount_ = _inputChannel ;//inputLayer->channels();
307+
this->inputGeometry_.height =inputHeight_;
308+
this->inputGeometry_.width = inputWidth_ ;//inputLayer->width();
309+
this->inputChannelCount_ = inputChannel_ ;//inputLayer->channels();
310310

311311
#else
312312
CV_Error(Error::StsError,"DNN module not available during compilation!");

0 commit comments

Comments
 (0)