Этот код является обучающим кодом для распознавания лиц с использованием OpenCV. Он обрезает и выравнивает изображения, чтобы включить только лицо.
Единственная возникающая ошибка — функция stat () возвращает значение -1, а файлы в D: \ Project \ Original не доступны.
Результат следующий:
-1 Не удается получить доступ к папке1
Пожалуйста, помогите мне понять, почему это происходит. Эта ошибка не возникала, когда тот же код (с небольшими изменениями) был скомпилирован в CodeBlocks 16.01 с OpenCV 2.4.9
#include <iostream>
#include <fcntl.h>
#include <sys/stat.h>
#include <string>
#include <string.h>
#include <fstream>
#include <string.h>
#include <direct.h>
#include <dirent.h>
#include <opencv/cv.h>
#include <opencv2/core.hpp>
#include <opencv2/highgui.hpp>
#include <opencv2/xfeatures2d.hpp>
#include <opencv2/imgproc.hpp>
#include <opencv2/objdetect.hpp>
using namespace std;
using namespace cv;
char* IMG_SEP = "_";
int MAX_IMAGE_NAME_SIZE = 100;
int IM_HEIGHT = 200;
int IM_WIDTH = 200;
String face_cascade_name = "haarcascade_frontalface_alt.xml";
String eyes_cascade_name = "haarcascade_eye_tree_eyeglasses.xml";
CascadeClassifier face_cascade;
CascadeClassifier eyes_cascade;
vector<string> fileNames;
vector<string> outputLabel;
vector<string> outputPath;
char* srcDirName = "D:/Project/Original";
char* dstDirName = "D:/Project/Cropped";
char* CSV_PATH = "D:/Project/Cropped/myCSV.csv";
void getFileNames(char* path) {
DIR *dir;
struct stat info;
struct dirent *ent;
if ((dir = opendir(path)) != NULL) {
while ((ent = readdir(dir)) != NULL) {
if ((!strcmp(ent->d_name, ".") == 0) && (!strcmp(ent->d_name, "..") == 0)) {
char* temp = (char*)malloc(256);
sprintf_s(temp, 100, "%s%s", path, ent->d_name);
// THIS IS WHERE THE DEBUGGER GOES. STAT RETURNS -1
if (stat(temp, &info) != 0) {
cout << stat(temp, &info) << " ";
cout << "Cannot access " << ent->d_name << "\n";
}if (info.st_mode & S_IFREG) {
string temp1 = ent->d_name;
temp1 = temp1.substr(0, temp1.rfind(IMG_SEP));
temp1 = dstDirName + temp1;
_mkdir(temp1.c_str());
fileNames.push_back(temp);
}
else if (info.st_mode & S_IFDIR) {
char* temp1 = (char*)malloc(256);
sprintf_s(temp1, 100, "%s/%s", dstDirName, ent->d_name);
_mkdir(temp1);
strcat_s(temp, sizeof temp, "/");
getFileNames(temp);
}
}
}
closedir(dir);
}
else {
cout << "Could not open the required image directory";
}
}
void printVector(vector<string> images) {
for (int i = 0; i < images.size(); i++) {
cout << images[i] << "\n";
}
}
void initializeDetector() {
if (!face_cascade.load(face_cascade_name)) {
cout << "Error while loading the face detection cascade classifier";
}
if (!eyes_cascade.load(eyes_cascade_name)) {
cout << "Error while loading the eye detection cascade classifier";
}
}
void detectAndStore(Mat image, String imageName) {
vector<Rect> faces;
vector<Rect> eyes;
Mat frame_grey;
cvtColor(image, frame_grey, CV_BGR2GRAY);
equalizeHist(frame_grey, frame_grey);
face_cascade.detectMultiScale(frame_grey, faces, 1.1, 2, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));
//if(faces.size() == 1){
for (size_t i = 0; i < faces.size(); i++) {
Mat faceROI = frame_grey(faces[i]);
eyes_cascade.detectMultiScale(faceROI, eyes, 1.1, 2, 0 | CV_HAAR_SCALE_IMAGE, Size(30, 30));
if (eyes.size() < 1) {
continue;
}
resize(faceROI, faceROI, Size(IM_WIDTH, IM_HEIGHT), 1.0, 1.0, INTER_CUBIC);
String fileName = imageName;
int posBeg = fileName.rfind('/') + 1;
String label = fileName.substr(posBeg, MAX_IMAGE_NAME_SIZE);
int posEnd = label.rfind('.');
String labelNameWithNum = label.substr(0, posEnd);
cout << "Cropping and storing " << labelNameWithNum << "\n";
int pos = labelNameWithNum.rfind(IMG_SEP);
String labelNameWithoutNum = labelNameWithNum.substr(0, pos);
String destination = dstDirName + labelNameWithoutNum + "/" + label;
outputPath.push_back(destination);
outputLabel.push_back(labelNameWithoutNum);
equalizeHist(faceROI, faceROI);
imwrite(destination, faceROI);
break;
}
}
void processImages() {
initializeDetector();
for (int i = 0; i < fileNames.size(); i++) {
Mat image = imread(fileNames[i], CV_LOAD_IMAGE_COLOR);
detectAndStore(image, fileNames[i]);
}
}
void createCSV() {
ofstream myFile;
myFile.open(CSV_PATH, ios_base::out);
for (int i = 0; i < outputPath.size(); i++) {
myFile << outputPath[i] << ";" << outputLabel[i] << "\n";
}
myFile.close();
}
int main() {
getFileNames(srcDirName);
processImages();
createCSV();
return 0;
}
Поскольку вы используете OpenCV 3.1, вы можете использовать шарик. Это позволяет избежать сложного кода, подобного вашему, вместе с его ошибкой.
Ваш getFileNames
сводится в основном к:
vector<cv::String> fileNames; // This must be a cv::String now
...
void getFileNames(std::string path)
{
cv::glob(path,fileNames,false);
}
Ты можешь видеть glob
используется во многих других ответах, например Вот.
sprintf_s(temp, 100, "%s%s", path, ent->d_name);
->
sprintf_s(temp, 100, "%s/%s", path, ent->d_name);
(добавить косую черту)
Это было тривиально отлажено путем проверки «temp» в точке несоответствующей статистики.
Так как dirent не входит в MSVC, возможно, это ошибка в dirent.h, которую вы вставили в … но вы делаете% s /% s позже с ent-> d_name, а «путь» происходит из вашей жестко закодированной константы, Я думаю, что вы сделали ошибку при портировании из «блоков кода»