File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -145,11 +145,11 @@ class DictNetCaffeImpl: public DictNet{
145
145
std::vector<Mat> allImageVector;
146
146
inputImageList.getMatVector (allImageVector);
147
147
size_t outputSize=size_t (this ->outputSize_ );// temporary variable to avoid int to size_t arithmentic
148
- size_t minibatchSize=this ->minibatchSz_ ;// temporary variable to avoid int to size_t arithmentic
149
- classProbabilities.create (Size (outputSize, allImageVector.size ()),CV_32F);
148
+ size_t minibatchSize=size_t ( this ->minibatchSz_ ) ;// temporary variable to avoid int to size_t arithmentic
149
+ classProbabilities.create (Size (int ( outputSize), int ( allImageVector.size () )),CV_32F);
150
150
Mat outputMat = classProbabilities.getMat ();
151
151
for (size_t imgNum=0 ;imgNum<allImageVector.size ();imgNum+=minibatchSize){
152
- int rangeEnd=imgNum+std::min<size_t >(allImageVector.size ()-imgNum,minibatchSize);
152
+ size_t rangeEnd=imgNum+std::min<size_t >(allImageVector.size ()-imgNum,minibatchSize);
153
153
std::vector<Mat>::const_iterator from=allImageVector.begin ()+imgNum;
154
154
std::vector<Mat>::const_iterator to=allImageVector.begin ()+rangeEnd;
155
155
std::vector<Mat> minibatchInput (from,to);
You can’t perform that action at this time.
0 commit comments