check for 'DOTjpg' and 'DOTpng'

severin-lemaignan · Nov 9, 2017 · d51a431 · d51a431
1 parent 24bee91
commit d51a431
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tools/estimate_head_pose_from_image_or_file.cpp b/tools/estimate_head_pose_from_image_or_file.cpp
@@ -117,7 +117,7 @@ int main(int argc, char **argv)
     auto t_start = getTickCount();
 
     // Single image file
-    if (fileName.find("jpg") != std::string::npos or fileName.find("png") != std::string::npos) {
+    if (fileName.find(".jpg") != std::string::npos or fileName.find(".png") != std::string::npos) {
         cout << ".jpg or .png file" << endl;
         estimate_head_pose_on_frameFileName(fileName, estimator, prev_poses, false);
     }