m53297601
/
EasyPR
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Merge branch 'master-gbk'

Browse Source
v1.6alpha
Micooz 10 years ago
parent 4e4396542d 0789f108df
commit 738d4bb48c
9 changed files with 133 additions and 74 deletions
Show Stats Download Patch File Download Diff File

21
include/easypr/util.h
Unescape View File

@ -4,6 +4,15 @@
#include <iostream> #include <iostream>
#include <string> #include <string>
#include <vector> #include <vector>
#include <opencv2/core/core.hpp>
#if defined(WIN32) || defined(_WIN32)
#define OS_WINDOWS
#elif defined(__APPLE__) || defined(APPLE)
#define OS_UNIX
#elif defined(__linux__) || defined(linux)
#define OS_LINUX
#endif
#ifdef min #ifdef min
#undef min #undef min
@ -89,9 +98,21 @@ class Utils {
return prevCol[len2]; return prevCol[len2];
} }
/*
* Create multi-level directories by given folder.
*/
static bool mkdir(const std::string folder);
/*
* Make sure the destination folder exists,
* if not, create it, then call cv::imwrite.
*/
static bool imwrite(const std::string& file, const cv::Mat& image);
}; };
typedef Utils utils;
} // namespace easypr } // namespace easypr
/// general_test_prepare.cpp中方法 /// general_test_prepare.cpp中方法

15
src/core/chars_segment.cpp
Unescape View File

@ -1,4 +1,5 @@
#include "easypr/chars_segment.h" #include "easypr/chars_segment.h"
#include "easypr/util.h"
/*! \namespace easypr /*! \namespace easypr
Namespace where all the C++ EasyPR functionality resides Namespace where all the C++ EasyPR functionality resides
@ -101,7 +102,7 @@ namespace easypr{
int h = input_grey.rows; int h = input_grey.rows;
Mat tmp = input_grey(Rect(w*0.1,h*0.1,w*0.8,h*0.8)); Mat tmp = input_grey(Rect(w*0.1,h*0.1,w*0.8,h*0.8));
int threadHoldV = ThresholdOtsu(tmp); int threadHoldV = ThresholdOtsu(tmp);
imwrite("./tmp/inputgray2.jpg",input_grey); utils::imwrite("tmp/inputgray2.jpg",input_grey);
threshold(input_grey, img_threshold,threadHoldV, 255, CV_THRESH_BINARY); threshold(input_grey, img_threshold,threadHoldV, 255, CV_THRESH_BINARY);
@ -116,7 +117,7 @@ namespace easypr{
int h = input_grey.rows; int h = input_grey.rows;
Mat tmp = input_grey(Rect(w*0.1,h*0.1,w*0.8,h*0.8)); Mat tmp = input_grey(Rect(w*0.1,h*0.1,w*0.8,h*0.8));
int threadHoldV = ThresholdOtsu(tmp); int threadHoldV = ThresholdOtsu(tmp);
imwrite("./tmp/inputgray2.jpg",input_grey); utils::imwrite("tmp/inputgray2.jpg",input_grey);
threshold(input_grey, img_threshold,threadHoldV, 255, CV_THRESH_BINARY_INV); threshold(input_grey, img_threshold,threadHoldV, 255, CV_THRESH_BINARY_INV);
@ -132,7 +133,7 @@ namespace easypr{
{ {
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << "tmp/debug_char_threshold" <<iTag<< ".jpg"; ss << "tmp/debug_char_threshold" <<iTag<< ".jpg";
imwrite(ss.str(), img_threshold); utils::imwrite(ss.str(), img_threshold);
} }
//去除车牌上方的柳钉以及下方的横线等干扰 //去除车牌上方的柳钉以及下方的横线等干扰
@ -145,7 +146,7 @@ namespace easypr{
{ {
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << "tmp/debug_char_clearLiuDing" <<iTag<< ".jpg"; ss << "tmp/debug_char_clearLiuDing" <<iTag<< ".jpg";
imwrite(ss.str(), img_threshold); utils::imwrite(ss.str(), img_threshold);
} }
iTag++; iTag++;
@ -200,7 +201,7 @@ namespace easypr{
Mat specMat(img_threshold, sortedRect[specIndex]); Mat specMat(img_threshold, sortedRect[specIndex]);
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << "tmp/debug_specMat" << ".jpg"; ss << "tmp/debug_specMat" << ".jpg";
imwrite(ss.str(), specMat); utils::imwrite(ss.str(), specMat);
} }
} }
@ -218,7 +219,7 @@ namespace easypr{
Mat chineseMat(img_threshold, chineseRect); Mat chineseMat(img_threshold, chineseRect);
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << "tmp/debug_chineseMat" << ".jpg"; ss << "tmp/debug_chineseMat" << ".jpg";
imwrite(ss.str(), chineseMat); utils::imwrite(ss.str(), chineseMat);
} }
@ -245,7 +246,7 @@ namespace easypr{
{ {
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << "tmp/debug_char_auxRoi_" << (i+staticIndex) << ".jpg"; ss << "tmp/debug_char_auxRoi_" << (i+staticIndex) << ".jpg";
imwrite(ss.str(), auxRoi); utils::imwrite(ss.str(), auxRoi);
} }
resultVec.push_back(auxRoi); resultVec.push_back(auxRoi);
} }

3
src/core/plate_detect.cpp
Unescape View File

@ -1,4 +1,5 @@
#include "easypr/plate_detect.h" #include "easypr/plate_detect.h"
#include "easypr/util.h"
/*! \namespace easypr /*! \namespace easypr
Namespace where all the C++ EasyPR functionality resides Namespace where all the C++ EasyPR functionality resides
@ -38,7 +39,7 @@ int CPlateDetect::plateDetect(Mat src, vector<Mat>& resultVec, int index) {
if (1) { if (1) {
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << "tmp/plate_judge_result_" << i << ".jpg"; ss << "tmp/plate_judge_result_" << i << ".jpg";
imwrite(ss.str(), img); utils::imwrite(ss.str(), img);
} }
} }
} }

45
src/core/plate_locate.cpp
Unescape View File

@ -1,4 +1,5 @@
#include "easypr/plate_locate.h" #include "easypr/plate_locate.h"
#include "easypr/util.h"
/*! \namespace easypr /*! \namespace easypr
Namespace where all the C++ EasyPR functionality resides Namespace where all the C++ EasyPR functionality resides
@ -90,7 +91,7 @@ Mat CPlateLocate::showResultMat(Mat src, Size rect_size, Point2f center,
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_crop_" << index << ".jpg"; ss << "tmp/debug_crop_" << index << ".jpg";
imwrite(ss.str(), img_crop); utils::imwrite(ss.str(), img_crop);
} }
Mat resultResized; Mat resultResized;
@ -101,7 +102,7 @@ Mat CPlateLocate::showResultMat(Mat src, Size rect_size, Point2f center,
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_resize_" << index << ".jpg"; ss << "tmp/debug_resize_" << index << ".jpg";
imwrite(ss.str(), resultResized); utils::imwrite(ss.str(), resultResized);
} }
return resultResized; return resultResized;
@ -134,7 +135,7 @@ int CPlateLocate::colorSearch(const Mat& src, const Color r, Mat& out,
morphologyEx(src_threshold, src_threshold, MORPH_CLOSE, element); morphologyEx(src_threshold, src_threshold, MORPH_CLOSE, element);
imwrite("./tmp/color.jpg", src_threshold); utils::imwrite("tmp/color.jpg", src_threshold);
src_threshold.copyTo(out); src_threshold.copyTo(out);
@ -245,7 +246,7 @@ int CPlateLocate::sobelFrtSearch(const Mat& src,
m_MorphSizeHeight); m_MorphSizeHeight);
imwrite("tmp/sobelFrtSearch.jpg", src_threshold); utils::imwrite("tmp/sobelFrtSearch.jpg", src_threshold);
vector<vector<Point> > contours; vector<vector<Point> > contours;
@ -322,14 +323,14 @@ int CPlateLocate::sobelSecSearchPart(Mat& bound, Point2f refpoint,
} }
imwrite("tmp/repaireimg1.jpg", bound_threshold); utils::imwrite("tmp/repaireimg1.jpg", bound_threshold);
//两边的区域不要 //两边的区域不要
for (int i = 0; i < bound_threshold.rows; i++) { for (int i = 0; i < bound_threshold.rows; i++) {
bound_threshold.data[i * bound_threshold.cols + posLeft] = 0; bound_threshold.data[i * bound_threshold.cols + posLeft] = 0;
bound_threshold.data[i * bound_threshold.cols + posRight] = 0; bound_threshold.data[i * bound_threshold.cols + posRight] = 0;
} }
imwrite("tmp/repaireimg2.jpg", bound_threshold); utils::imwrite("tmp/repaireimg2.jpg", bound_threshold);
} }
@ -373,7 +374,7 @@ int CPlateLocate::sobelSecSearch(Mat& bound, Point2f refpoint,
//! 第二次参数比一次精细，但针对的是得到的外接矩阵之后的图像，再sobel得到二值图像 //! 第二次参数比一次精细，但针对的是得到的外接矩阵之后的图像，再sobel得到二值图像
sobelOper(bound, bound_threshold, 3, 10, 3); sobelOper(bound, bound_threshold, 3, 10, 3);
imwrite("./tmp/sobelSecSearch.jpg", bound_threshold); utils::imwrite("tmp/sobelSecSearch.jpg", bound_threshold);
vector<vector<Point> > contours; vector<vector<Point> > contours;
findContours(bound_threshold, findContours(bound_threshold,
@ -472,7 +473,7 @@ void DeleteNotArea(Mat& inmat) {
threshold(input_grey, img_threshold, threadHoldV, 255, CV_THRESH_BINARY); threshold(input_grey, img_threshold, threadHoldV, 255, CV_THRESH_BINARY);
//threshold(input_grey, img_threshold, 5, 255, CV_THRESH_OTSU + CV_THRESH_BINARY); //threshold(input_grey, img_threshold, 5, 255, CV_THRESH_OTSU + CV_THRESH_BINARY);
imwrite("./tmp/inputgray2.jpg", img_threshold); utils::imwrite("tmp/inputgray2.jpg", img_threshold);
} }
else if (YELLOW == plateType) { else if (YELLOW == plateType) {
@ -1024,7 +1025,7 @@ int CPlateLocate::sobelOperT(const Mat& in, Mat& out, int blurSize, int morphW,
//mat_gray = m_bilateral.BilateralFilter(mat_gray,6); //mat_gray = m_bilateral.BilateralFilter(mat_gray,6);
imwrite("./tmp/grayblure.jpg", mat_gray); utils::imwrite("tmp/grayblure.jpg", mat_gray);
//equalizeHist(mat_gray, mat_gray); //equalizeHist(mat_gray, mat_gray);
@ -1044,18 +1045,18 @@ int CPlateLocate::sobelOperT(const Mat& in, Mat& out, int blurSize, int morphW,
Mat grad; Mat grad;
addWeighted(abs_grad_x, 1, abs_grad_y, 0, 0, grad); addWeighted(abs_grad_x, 1, abs_grad_y, 0, 0, grad);
imwrite("./tmp/graygrad.jpg", grad); utils::imwrite("tmp/graygrad.jpg", grad);
Mat mat_threshold; Mat mat_threshold;
//double otsu_thresh_val = //double otsu_thresh_val =
threshold(grad, mat_threshold, 0, 255, CV_THRESH_OTSU + CV_THRESH_BINARY); threshold(grad, mat_threshold, 0, 255, CV_THRESH_OTSU + CV_THRESH_BINARY);
imwrite("./tmp/grayBINARY.jpg", mat_threshold); utils::imwrite("tmp/grayBINARY.jpg", mat_threshold);
Mat element = getStructuringElement(MORPH_RECT, Size(morphW, morphH)); Mat element = getStructuringElement(MORPH_RECT, Size(morphW, morphH));
morphologyEx(mat_threshold, mat_threshold, MORPH_CLOSE, element); morphologyEx(mat_threshold, mat_threshold, MORPH_CLOSE, element);
imwrite("./tmp/phologyEx.jpg", mat_threshold); utils::imwrite("tmp/phologyEx.jpg", mat_threshold);
out = mat_threshold; out = mat_threshold;
@ -1175,7 +1176,7 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_GaussianBlur" << ".jpg"; ss << "tmp/debug_GaussianBlur" << ".jpg";
imwrite(ss.str(), src_blur); utils::imwrite(ss.str(), src_blur);
} }
/// Convert it to gray /// Convert it to gray
@ -1184,7 +1185,7 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_gray" << ".jpg"; ss << "tmp/debug_gray" << ".jpg";
imwrite(ss.str(), src_gray); utils::imwrite(ss.str(), src_gray);
} }
// RGB颜色初定位 // RGB颜色初定位
// http://wenku.baidu.com/view/2329e5d2360cba1aa811da65.html?re=view // http://wenku.baidu.com/view/2329e5d2360cba1aa811da65.html?re=view
@ -1330,9 +1331,9 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_Sobel_blue" << ".jpg"; ss << "tmp/debug_Sobel_blue" << ".jpg";
imwrite(ss.str(), out_blue); utils::imwrite(ss.str(), out_blue);
ss << "tmp/debug_Sobel_yellow" << ".jpg"; ss << "tmp/debug_Sobel_yellow" << ".jpg";
imwrite(ss.str(), out_yellow); utils::imwrite(ss.str(), out_yellow);
} }
Mat img_threshold_blue; Mat img_threshold_blue;
@ -1346,9 +1347,9 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_threshold_blue" << ".jpg"; ss << "tmp/debug_threshold_blue" << ".jpg";
imwrite(ss.str(), img_threshold_blue); utils::imwrite(ss.str(), img_threshold_blue);
ss << "tmp/debug_threshold_yellow" << ".jpg"; ss << "tmp/debug_threshold_yellow" << ".jpg";
imwrite(ss.str(), img_threshold_yellow); utils::imwrite(ss.str(), img_threshold_yellow);
} }
Mat element = getStructuringElement(MORPH_RECT, Size(m_MorphSizeWidth, Mat element = getStructuringElement(MORPH_RECT, Size(m_MorphSizeWidth,
@ -1360,9 +1361,9 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_morphology_blue" << ".jpg"; ss << "tmp/debug_morphology_blue" << ".jpg";
imwrite(ss.str(), img_threshold_blue); utils::imwrite(ss.str(), img_threshold_blue);
ss << "tmp/debug_morphology_yellow" << ".jpg"; ss << "tmp/debug_morphology_yellow" << ".jpg";
imwrite(ss.str(), img_threshold_yellow); utils::imwrite(ss.str(), img_threshold_yellow);
} }
//Find 轮廓 of possibles plates //Find 轮廓 of possibles plates
@ -1391,7 +1392,7 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
1); // with a thickness of 1 1); // with a thickness of 1
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_Contours" << ".jpg"; ss << "tmp/debug_Contours" << ".jpg";
imwrite(ss.str(), result); utils::imwrite(ss.str(), result);
} }
@ -1471,7 +1472,7 @@ int CPlateLocate::plateLocate(Mat src, vector<Mat>& resultVec, int index) {
if (m_debug) { if (m_debug) {
stringstream ss(stringstream::in | stringstream::out); stringstream ss(stringstream::in | stringstream::out);
ss << "tmp/debug_result" << ".jpg"; ss << "tmp/debug_result" << ".jpg";
imwrite(ss.str(), result); utils::imwrite(ss.str(), result);
} }
return 0; return 0;

2
src/preprocess/gdts.cpp
Unescape View File

@ -55,7 +55,7 @@ int generate_gdts() {
std::stringstream ss(std::stringstream::in | std::stringstream::out); std::stringstream ss(std::stringstream::in | std::stringstream::out);
ss << dst_path << "/" << filename; ss << dst_path << "/" << filename;
imwrite(ss.str(), dst); utils::imwrite(ss.str(), dst);
} }
return 0; return 0;

9
src/preprocess/mc_data.cpp
Unescape View File

@ -5,13 +5,12 @@
#include <ctime> #include <ctime>
#include "easypr/plate_judge.h" #include "easypr/plate_judge.h"
#include "easypr/plate_locate.h" #include "easypr/plate_locate.h"
#include "easypr/util.h"
#if defined (WIN32) || defined (_WIN32) #ifdef OS_WINDOWS
#include <io.h> #include <io.h>
#endif #endif
#include "easypr/util.h"
namespace easypr { namespace easypr {
namespace preprocess { namespace preprocess {
@ -115,7 +114,7 @@ void create_learn_data(const char* raw_data_folder, const char* out_data_folder,
} }
save_to.append(Utils::getFileName(f, true)); save_to.append(Utils::getFileName(f, true));
cv::imwrite(save_to, img); utils::imwrite(save_to, img);
std::cout << f << " -> " << save_to << std::endl; std::cout << f << " -> " << save_to << std::endl;
} }
std::cout << "Learn data created successfully!" << std::endl; std::cout << "Learn data created successfully!" << std::endl;
@ -169,7 +168,7 @@ void tag_data(const char* source_folder, const char* has_plate_folder,
no_plate_folder, filename.c_str(), plate_index); no_plate_folder, filename.c_str(), plate_index);
std::cout << "[N] -> " << save_to << std::endl; std::cout << "[N] -> " << save_to << std::endl;
} }
cv::imwrite(save_to, plate); utils::imwrite(save_to, plate);
++plate_index; ++plate_index;
} }
} }

16
src/train/ann_train.cpp
Unescape View File

@ -1,11 +1,10 @@
// ann_train.cpp : ann模型的训练文件，主要用在OCR中 // ann_train.cpp : ann模型的训练文件，主要用在OCR中
#include <opencv/cv.h> #include <opencv2/opencv.hpp>
#include <opencv/cvaux.h> #include "easypr/util.h"
#include "opencv2/ml/ml.hpp" #include "easypr/plate_recognize.h"
#include "opencv2/highgui/highgui.hpp"
#if defined (WIN32) || defined (_WIN32) #ifdef OS_WINDOWS
#include <windows.h> #include <windows.h>
#endif #endif
@ -13,15 +12,10 @@
#include <iostream> #include <iostream>
#include <cstdlib> #include <cstdlib>
#if defined (WIN32) || defined (_WIN32) #ifdef OS_WINDOWS
#include <io.h> #include <io.h>
#elif defined (linux) || defined (__linux__)
//#include <sys/io.h>
#endif #endif
#include "easypr/plate_recognize.h"
#include "easypr/util.h"
using namespace easypr; using namespace easypr;
using namespace cv; using namespace cv;
using namespace std; using namespace std;

2
src/train/svm_train.cpp
Unescape View File

@ -34,7 +34,7 @@ void Svm::divide(const char* images_folder, float percentage /* = 0.7 */) {
} else { } else {
sprintf(save_to, "%s/test/%s", images_folder, file_name); sprintf(save_to, "%s/test/%s", images_folder, file_name);
} }
cv::imwrite(save_to, image); utils::imwrite(save_to, image);
std::cout << f << " -> " << save_to << std::endl; std::cout << f << " -> " << save_to << std::endl;
} }
} }

94
src/util/util.cpp
Unescape View File

@ -1,33 +1,37 @@
#include "easypr/util.h" #include "easypr/util.h"
#if defined(WIN32) || defined(_WIN32) #ifdef OS_WINDOWS
#include <windows.h> #include <windows.h>
#include <direct.h>
#include <io.h> #include <io.h>
#elif defined(linux) || defined(__linux__) || defined(__APPLE__) #define PATH_DELIMITER '\\'
#elif defined(OS_LINUX) || defined(OS_UNIX)
#include <cstring> #include <cstring>
#include <dirent.h> #include <dirent.h>
#include <sys/stat.h> #include <sys/stat.h>
#include <unistd.h>
#define PATH_DELIMITER '/'
#endif #endif
#if defined(__APPLE__) #ifdef OS_UNIX
#include <sys/timeb.h> #include <sys/timeb.h>
#endif #endif
#include <list> #include <list>
#include <opencv2/highgui/highgui.hpp>
using namespace std; namespace easypr {
using namespace easypr;
long Utils::getTimestamp() { long Utils::getTimestamp() {
#if defined(WIN32) || defined(_WIN32) #ifdef OS_WINDOWS
return GetTickCount(); return GetTickCount();
#endif #endif
#if (linux) || defined(__linux__) #ifdef OS_LINUX
struct timespec ts; struct timespec ts;
clock_gettime(CLOCK_MONOTONIC, &ts); clock_gettime(CLOCK_MONOTONIC, &ts);
@ -35,7 +39,7 @@ long Utils::getTimestamp() {
return (ts.tv_sec * 1e3 + ts.tv_nsec / 1e6); return (ts.tv_sec * 1e3 + ts.tv_nsec / 1e6);
#endif #endif
#if defined(__APPLE__) #ifdef OS_UNIX
// there is no function provided by osx to get system tick count. // there is no function provided by osx to get system tick count.
// but considering the purpose by using this function, // but considering the purpose by using this function,
// we can simply return a millisecond since 1970/1/1 to calc the time elapse. // we can simply return a millisecond since 1970/1/1 to calc the time elapse.
@ -45,10 +49,10 @@ long Utils::getTimestamp() {
#endif #endif
} }
std::string Utils::getFileName(const string& path, std::string Utils::getFileName(const std::string& path,
const bool postfix /* = false */) { const bool postfix /* = false */) {
if (!path.empty()) { if (!path.empty()) {
#if defined(WIN32) || defined(_WIN32) #ifdef OS_WINDOWS
size_t last_slash_1 = path.find_last_of("\\"); size_t last_slash_1 = path.find_last_of("\\");
size_t last_slash_2 = path.find_last_of("/"); size_t last_slash_2 = path.find_last_of("/");
size_t last_slash; size_t last_slash;
@ -68,7 +72,7 @@ std::string Utils::getFileName(const string& path,
#endif #endif
size_t last_dot = path.find_last_of('.'); size_t last_dot = path.find_last_of('.');
if (last_dot < last_slash || last_dot == string::npos) { if (last_dot < last_slash || last_dot == std::string::npos) {
// not found the right dot of the postfix, // not found the right dot of the postfix,
// return the file name directly // return the file name directly
return path.substr(last_slash + 1); return path.substr(last_slash + 1);
@ -85,13 +89,14 @@ std::string Utils::getFileName(const string& path,
return ""; return "";
} }
vector<string> Utils::splitString(const string& str, const char delimiter) { std::vector<std::string> Utils::splitString(const std::string& str,
vector<string> splited; const char delimiter) {
string s(str); std::vector<std::string> splited;
std::string s(str);
size_t pos; size_t pos;
while ((pos = s.find(delimiter)) != string::npos) { while ((pos = s.find(delimiter)) != std::string::npos) {
string sec = s.substr(0, pos); std::string sec = s.substr(0, pos);
if (!sec.empty()) { if (!sec.empty()) {
splited.push_back(s.substr(0, pos)); splited.push_back(s.substr(0, pos));
@ -105,12 +110,12 @@ vector<string> Utils::splitString(const string& str, const char delimiter) {
return splited; return splited;
} }
vector<string> Utils::getFiles(const string& folder, std::vector<std::string> Utils::getFiles(const std::string& folder,
const bool all /* = true */) { const bool all /* = true */) {
vector<string> files; std::vector<std::string> files;
list<string> subfolders; std::list<std::string> subfolders;
subfolders.push_back(folder); subfolders.push_back(folder);
#if defined(WIN32) || defined(_WIN32) #ifdef OS_WINDOWS
while (!subfolders.empty()) { while (!subfolders.empty()) {
string current_folder(subfolders.back()); string current_folder(subfolders.back());
@ -156,9 +161,9 @@ vector<string> Utils::getFiles(const string& folder,
} // while } // while
_findclose(file_handler); _findclose(file_handler);
} }
#elif defined(linux) || defined(__linux__) || defined(__APPLE__) #elif defined(OS_LINUX) || defined(OS_UNIX)
while (!subfolders.empty()) { while (!subfolders.empty()) {
string current_folder(subfolders.back()); std::string current_folder(subfolders.back());
if (*(current_folder.end() - 1) != '/') { if (*(current_folder.end() - 1) != '/') {
current_folder.push_back('/'); current_folder.push_back('/');
@ -188,7 +193,7 @@ vector<string> Utils::getFiles(const string& folder,
continue; continue;
} }
string file_path; std::string file_path;
file_path.append(current_folder.c_str()); file_path.append(current_folder.c_str());
file_path.append(dir->d_name); file_path.append(dir->d_name);
@ -202,7 +207,7 @@ vector<string> Utils::getFiles(const string& folder,
// it's a sub folder // it's a sub folder
if (all) { if (all) {
// will search sub folder // will search sub folder
string subfolder(current_folder); std::string subfolder(current_folder);
subfolder.append(dir->d_name); subfolder.append(dir->d_name);
subfolders.push_back(subfolder.c_str()); subfolders.push_back(subfolder.c_str());
@ -214,7 +219,44 @@ vector<string> Utils::getFiles(const string& folder,
} // while } // while
closedir(pdir); closedir(pdir);
} }
#endif #endif
return files; return files;
} }
bool Utils::mkdir(const std::string folder) {
std::string folder_builder;
std::string sub;
sub.reserve(folder.size());
for (auto it = folder.begin(); it != folder.end(); ++it) {
const char c = *it;
sub.push_back(c);
if (c == PATH_DELIMITER || it == folder.end() - 1) {
folder_builder.append(sub);
#ifdef OS_WINDOWS
if (0 != ::_access(folder_builder.c_str(), 0)) {
#else
if (0 != ::access(folder_builder.c_str(), 0)) {
#endif
// this folder not exist
#ifdef OS_WINDOWS
if (0 != ::_mkdir(folder_builder.c_str())) {
#else
if (0 != ::mkdir(folder_builder.c_str(), S_IRWXU)) {
#endif
// create failed
return false;
}
}
sub.clear();
}
}
return true;
}
bool Utils::imwrite(const std::string& file, const cv::Mat& image) {
auto folder = file.substr(0, file.find_last_of(PATH_DELIMITER));
Utils::mkdir(folder);
return cv::imwrite(file, image);
}
} // namespace easypr
Write Preview
Loading…
Cancel
Save