From d51a431299b0b8cf53755edb6f2098a57ad6249b Mon Sep 17 00:00:00 2001
From: Vikram Voleti <vikky2904@gmail.com>
Date: Thu, 9 Nov 2017 20:53:39 +0530
Subject: [PATCH] check for 'DOTjpg' and 'DOTpng'

---
 tools/estimate_head_pose_from_image_or_file.cpp | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tools/estimate_head_pose_from_image_or_file.cpp b/tools/estimate_head_pose_from_image_or_file.cpp
index 677fed1..bd62dd5 100644
--- 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);
     }