dockerfile/openalpr-base
1
0
Fork 0
mirror of https://github.com/kerberos-io/openalpr-base.git synced 2025-10-07 16:21:01 +08:00
Code Issues Packages Projects Releases Wiki Activity

Using a check function to see if image is supported

Browse Source
This commit is contained in:
Matt Hill
2015-05-13 07:07:09 -04:00
parent 3ac253b668
commit 98e66fc926
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
src/main.cpp
View File

@@ -42,6 +42,7 @@ const std::string LAST_VIDEO_STILL_LOCATION = "/tmp/laststill.jpg";
/** Function Headers */ /** Function Headers */
bool detectandshow(Alpr* alpr, cv::Mat frame, std::string region, bool writeJson); bool detectandshow(Alpr* alpr, cv::Mat frame, std::string region, bool writeJson);
bool is_supported_image(std::string image_file);
bool measureProcessingTime = false; bool measureProcessingTime = false;
std::string templatePattern; std::string templatePattern;
@@ -218,9 +219,7 @@ int main( int argc, const char** argv )
std::cerr << "Video file not found: " << filename << std::endl; std::cerr << "Video file not found: " << filename << std::endl;
} }
} }
else if (hasEndingInsensitive(filename, ".png") || hasEndingInsensitive(filename, ".jpg") || else if (is_supported_image(filename))
hasEndingInsensitive(filename, ".tif") || hasEndingInsensitive(filename, ".bmp") ||
hasEndingInsensitive(filename, ".jpeg") || hasEndingInsensitive(filename, ".gif"))
{ {
if (fileExists(filename.c_str())) if (fileExists(filename.c_str()))
{ {
@@ -244,7 +243,7 @@ int main( int argc, const char** argv )
for (int i = 0; i< files.size(); i++) for (int i = 0; i< files.size(); i++)
{ {
if (hasEndingInsensitive(files[i], ".jpg") || hasEndingInsensitive(files[i], ".png")) if (is_supported_image(files[i]))
{ {
std::string fullpath = filename + "/" + files[i]; std::string fullpath = filename + "/" + files[i];
std::cout << fullpath << std::endl; std::cout << fullpath << std::endl;
@@ -269,7 +268,12 @@ int main( int argc, const char** argv )
return 0; return 0;
} }
bool is_supported_image(std::string image_file)
{
return (hasEndingInsensitive(image_file, ".png") || hasEndingInsensitive(image_file, ".jpg") ||
hasEndingInsensitive(image_file, ".tif") || hasEndingInsensitive(image_file, ".bmp") ||
hasEndingInsensitive(image_file, ".jpeg") || hasEndingInsensitive(image_file, ".gif"));
}
bool detectandshow( Alpr* alpr, cv::Mat frame, std::string region, bool writeJson) bool detectandshow( Alpr* alpr, cv::Mat frame, std::string region, bool writeJson)