ZF/OpenFace
1
0
Fork 0
mirror of https://gitcode.com/gh_mirrors/ope/OpenFace.git synced 2026-05-15 03:37:49 +00:00
Code Issues Packages Projects Releases Wiki Activity

Bug fixes with image readers.

Browse Source
This commit is contained in:
Tadas Baltrusaitis
2017-11-19 20:04:00 +00:00
parent 974cee63b9
commit 488cd4d92f
3 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/local/Utilities/src/ImageCapture.cpp
View File

@@ -369,18 +369,19 @@ void ImageCapture::SetCameraIntrinsics(float fx, float fy, float cx, float cy)
cv::Mat ImageCapture::GetNextImage()
{
frame_num++;
if (image_files.empty() || frame_num - 1 > image_files.size())
if (image_files.empty() || frame_num >= image_files.size())
{
// Indicate lack of success by returning an empty image
latest_frame = cv::Mat();
return latest_frame;
}
latest_frame = cv::imread(image_files[frame_num - 1], -1);
latest_frame = cv::imread(image_files[frame_num], -1);
if (latest_frame.empty())
{
ERROR_STREAM("Could not open the image: " + image_files[frame_num - 1]);
exit(1);
}
image_height = latest_frame.size().height;
@@ -395,7 +396,9 @@ cv::Mat ImageCapture::GetNextImage()
// Set the grayscale frame
convertToGrayscale(latest_frame, latest_gray_frame);
this->name = boost::filesystem::path(image_files[frame_num - 1]).filename().replace_extension("").string();
this->name = boost::filesystem::path(image_files[frame_num]).filename().replace_extension("").string();
frame_num++;
return latest_frame;
}