OpenCVのimwriteがnullになる

Question

発生している問題

imshow で表示できている画像情報が imwrite で保存できません．

以下の機能を持つプログラムを作成しました．

1. ドラッグ&ドロップしたディレクトリ(inputDir)の直下にディレクトリ（outputDir）を作成
2. inputDir内の画像を読み込みリサイズし，outputDir 内に保存

しかしながら，現状，outputDir 内に空のファイルが作成されるのみとなっています．

また，複数の画像ファイルが存在しているディレクトリを入力しても，処理が 1 枚分で止まっているようです．

なので，正しく動作するためにはどう修正すればよいかお教え願います．

試したこと

• imwrite を除いた状態で imshow で dst を表示 -> 正常に処理される

環境

• Microsoft windows 10 64bit
• Visual Studio 2013 Ultimate
• OpenCV_3.0.0

参考にしたサイト

• フォルダ内の全ての画像ファイル名を取得する方法(C++)（http://tecsingularity.com/opencv/filename/）
• string に vector を使う（http://www.eonet.ne.jp/~maeda/cpp/string3.htm）
• OpenCVのimreadをしてもMat::dataがNULLになる（https://teratail.com/questions/45001）

プログラム

#define _CRT_SECURE_NO_WARNINGS

#include <Windows.h>
#include <direct.h>
#include <stdexcept>
#include <iostream>
#include <vector>
#include <string>
#include <opencv2/opencv.hpp>

using namespace std;
using namespace cv;

string inputDir;

vector<string> getImageName(string dir_name);

int main(int argc, char* argv[])
{
	// 入力ディレクトリの取得
	if (argc > 1) {
		inputDir = argv[1];
		cout << "input directory : " << inputDir << endl;
	}
	else {
		cerr << "no input file" << endl;
		while (1) {};
		return -1;
	}

	size_t length = inputDir.length();
	//cout << "length : " << length << endl;

	while (true) {
		auto pos = inputDir.find("\");
		if (pos == -1)	break;

		inputDir.replace(pos, 1, "/");
		//cout << "pos : " << pos + 1 << endl;
	}

	string dir = inputDir + "/";
	string outputDir = dir + "resized";

	try {
		if (_mkdir(outputDir.c_str()) == 0){
			cout << "success create dir : " << outputDir << endl;
		}
		else{
			cout << "do not create." << endl;
		}
	}
	catch (...) {
		cout << "input error" << endl;
		while (1) {};
	}

	// ファイル内の画像名の取得
	vector<string> file_names = getImageName(dir);
		
	// 表示・保存部分
	string filePath;
	for (auto f : file_names) {
		//cout << f << endl;
		
		string fileName = f.c_str();
		//cout << "fileName : " << fileName << endl;
		string filePath = dir + fileName;
		//cout << "filePath : " << filePath << endl;
		Mat src = imread(filePath);
		Mat dst;

		resize(src, dst, Size(227, 227), 0, 0, INTER_LINEAR);

		string outputPath = outputDir + "/" + fileName;
		cout << "output file path : " << outputPath << endl;

		imshow("img", dst);
		waitKey(30);

		imwrite(outputPath, dst);
		waitKey(30);
	}

	system("pause");
	return 0;
}

// inputDir内にある画像名を取得する
vector<string> getImageName(string dir_name) {

	HANDLE hFind;
	WIN32_FIND_DATA win32fd;
	std::vector<std::string> file_names;

	// png,jpg,bmpの拡張子のファイルのみを読み込む
	std::string extension[3] = { "png", "jpg", "bmp" };

	for (int i = 0; i < 3; i++) {

		string search_name = dir_name + "*." + extension[i];
		hFind = FindFirstFile(search_name.c_str(), &win32fd);

		if (hFind == INVALID_HANDLE_VALUE) {
			continue;
		}
		do {

			if (win32fd.dwFileAttributes & FILE_ATTRIBUTE_DIRECTORY) {
			}
			else {
				file_names.push_back(win32fd.cFileName);
			}
		} while (FindNextFile(hFind, &win32fd));

		FindClose(hFind);
	}

	return file_names;
}

Answer 1

こんにちは。

Visual Stuido 2017 + OpenCV 3.2でやってみましたが、問題なく動作しました。
ファイル名のパスに日本語が含まれていると正常動作しなことがあります。
jpgファイル等を日本語が含まれないパスに保存してトライしてみるとうまくいくかも知れません。

Answer 2

コードを読む限り、FindFirstFileの結果くらいしか疑うところは無さそうです。
パッと思いつく可能性としては、Unicode版が動作してて、win32fd.cFileNameがwstringになっている、など。

Answer 3

cv::imwriteのドキュメントのサンプルによると
cv::imwriteは例外を投げる可能性があるようです。

とりあえず、例外を記録してみましょう。

try {
    imwrite(outputPath, dst);
} catch (runtime_error& ex) {
    cerr << "Exception:" << ex.what() << endl;
    cerr << "at " << outputPath << endl;
}