Mercurial > hg > nsaunier > traffic-intelligence

comparison c/feature-based-tracking.cpp @ 534:d0419b1267dd

Find changesets by keywords (author, files, the commit message), revision number or hash, or revset expression.
fixing size issue when testing frames in main loop
author Nicolas Saunier <nicolas.saunier@polymtl.ca>
date Sun, 29 Jun 2014 14:37:30 -0400
parents e37f38274d4f
children 5ad2f51ae42f
comparison
equal deleted inserted replaced
533:e37f38274d4f 534:d0419b1267dd
157 lastFrameNum = MIN(params.frame1+static_cast<unsigned int>(params.nFrames), nFrames); 157 lastFrameNum = MIN(params.frame1+static_cast<unsigned int>(params.nFrames), nFrames);
158 158
159 capture->setFrameNumber(params.frame1); 159 capture->setFrameNumber(params.frame1);
160 for (unsigned int frameNum = params.frame1; (frameNum < lastFrameNum) && !::interruptionKey(key); frameNum++) { 160 for (unsigned int frameNum = params.frame1; (frameNum < lastFrameNum) && !::interruptionKey(key); frameNum++) {
161 bool success = capture->getNextFrame(frame); 161 bool success = capture->getNextFrame(frame);
162 if (!success || frame.empty() || frame.size() != videoSize) { 162 if (!success || frame.empty()) {
163 cout << "Empty frame " << frameNum << ", breaking (" << success << " " << frame.empty() << " [" << frame.size().width << "x" << frame.size().height << "])" << endl; 163 cout << "Empty frame " << frameNum << ", breaking (" << success << " " << frame.empty() << " [" << frame.size().width << "x" << frame.size().height << "])" << endl;
164 break; 164 break;
165 } else if (frameNum%50 ==0) 165 } else if (frameNum%50 ==0)
166 cout << "frame " << frameNum << endl; 166 cout << "frame " << frameNum << endl;
167 167
168 if (params.undistort) { 168 if (params.undistort) {
169 remap(frame, undistortedFrame, map1, map2, interpolationMethod, BORDER_CONSTANT, 0.); 169 remap(frame, undistortedFrame, map1, map2, interpolationMethod, BORDER_CONSTANT, 0.);
170 frame = undistortedFrame; 170 frame = undistortedFrame;
171
172 if (frame.size() != videoSize) {
173 cout << "Different frame size " << frameNum << ", breaking ([" << frame.size().width << "x" << frame.size().height << "])" << endl;
174 break;
175 }
171 } 176 }
172 177
173 178
174 cvtColor(frame, currentFrameBW, CV_RGB2GRAY); 179 cvtColor(frame, currentFrameBW, CV_RGB2GRAY);
175 180