Skip to content

Commit

Permalink
ノイズ除去レベル3を追加
Browse files Browse the repository at this point in the history
  • Loading branch information
lltcggie committed Apr 20, 2016
1 parent 1497e4f commit 77b8497
Show file tree
Hide file tree
Showing 18 changed files with 39 additions and 16 deletions.
11 changes: 6 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@
[Caffe](http://caffe.berkeleyvision.org/)を用いて書き直し、Windows向けにビルドしたソフトです。
CPUで変換することも出来ますが、CUDA(あるいはcuDNN)を使うとCPUより高速に変換することが出来ます。

GUI supports English and Japanese and Simplified Chinese and Korean.
GUI supports English and Japanese and Simplified Chinese and Traditional Chinese and Korean and Turkish.


要求環境
Expand Down Expand Up @@ -262,9 +262,9 @@ waifu2x-caffe-cui.exe -i mywaifu.png -m noise_scale --scale_ratio 1.6 --noise_le
画像の縦横比を維持したまま、指定された縦幅になるように拡大します(単位はピクセル)。
scale_widthと同時に指定することは出来ません。

###-n <1|2>, --noise_level <1|2>
ノイズ除去レベルを指定します。ノイズ除去用のモデルはレベル1とレベル2のみ用意されているので
1 もしくは 2 を指定して下さい。
###-n <1|2|3>, --noise_level <1|2|3>
ノイズ除去レベルを指定します。ノイズ除去用のモデルはレベル1とレベル2とレベル3のみ用意されているので
1 か 2 か 3 を指定して下さい。
デフォルト値は`1`です。

###--model_dir <文字列>
Expand Down Expand Up @@ -371,4 +371,5 @@ ex.
オリジナルのwaifu2xを元に[waifu2x-converter](https://github.com/WL-Amigo/waifu2x-converter-cpp)を作成して下さった [アミーゴ](https://twitter.com/WL_Amigo)さん(READMEやLICENSE.txtの書き方、OpenCVの使い方等かなり参考にさせていただきました)
に、感謝します。
また、メッセージを英訳してくださった @paul70078 さん、メッセージを中国語(簡体字)に翻訳してくださった @yoonhakcher さん、中国語(簡体字)訳のプルリクエストを下さった @mzhboy さん、
メッセージを韓国語に翻訳してくださった @kenin0726 さん、韓国語訳の改善を提案してくださった @aruhirin さん、メッセージを中国語(繁体字)に翻訳してくださった @lizardon1995 さん、@yoonhakcher さんに感謝します。
メッセージを韓国語に翻訳してくださった @kenin0726 さん、韓国語訳の改善を提案してくださった @aruhirin さん、
メッセージを中国語(繁体字)に翻訳してくださった @lizardon1995 さん、@yoonhakcher さん、トルコ語訳のプルリクエストを下さった @Scharynche さん に感謝します。
3 changes: 2 additions & 1 deletion bin/lang/Chinese_Simplified.json
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{
{
"IDC_STATIC_IO_SETTING":"输入和输出设置",
"IDC_STATIC_INPUT_PATH":"输入\r\n(文件或文件夹)",
"IDC_BUTTON_INPUT_REF":"选择",
Expand All @@ -20,6 +20,7 @@
"IDC_STATIC_JPEG_NOISE_LEVEL":"JPEG 降噪等级",
"IDC_RADIONOISE_LEVEL1":"1 级",
"IDC_RADIONOISE_LEVEL2":"2 级",
"IDC_RADIONOISE_LEVEL3":"3 级",
"IDC_STATIC_SCALE_RATE":"放大尺寸",
"IDC_RADIO_SCALE_RATIO":"放大倍率",
"IDC_RADIO_SCALE_WIDTH":"放大后宽度",
Expand Down
3 changes: 2 additions & 1 deletion bin/lang/Chinese_Traditional.json
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{
{
"IDC_STATIC_IO_SETTING":"輸入和輸出設置",
"IDC_STATIC_INPUT_PATH":"輸入路徑\r\n(可使用文件夾)",
"IDC_BUTTON_INPUT_REF":"瀏覽文件",
Expand All @@ -20,6 +20,7 @@
"IDC_STATIC_JPEG_NOISE_LEVEL":"JPEG降噪水平",
"IDC_RADIONOISE_LEVEL1":"等級1",
"IDC_RADIONOISE_LEVEL2":"等級2",
"IDC_RADIONOISE_LEVEL3":"等級3",
"IDC_STATIC_SCALE_RATE":"放大大小",
"IDC_RADIO_SCALE_RATIO":"放大比率",
"IDC_RADIO_SCALE_WIDTH":"轉換後高度",
Expand Down
4 changes: 2 additions & 2 deletions bin/lang/LangList.txt
Original file line number Diff line number Diff line change
@@ -1,6 +1,6 @@
;LangName LangID SubLangID FileName
;LangName LangID SubLangID FileName
English(US) 0x09 0x01 english.json
Türkçe(TR) 0x1F 0x01 turkish.json
Türkçe(TR) 0x1F 0x01 turkish.json
日本語 0x11 0x01 japanese.json
한국어 0x12 0x01 korean.json
繁体中文 0x04 0x01 Chinese_Traditional.json
Expand Down
1 change: 1 addition & 0 deletions bin/lang/english.json
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@
"IDC_STATIC_JPEG_NOISE_LEVEL":"JPEG denoise level",
"IDC_RADIONOISE_LEVEL1":"Level 1",
"IDC_RADIONOISE_LEVEL2":"Level 2",
"IDC_RADIONOISE_LEVEL3":"Level 3",
"IDC_STATIC_SCALE_RATE":"Magnification size",
"IDC_RADIO_SCALE_RATIO":"Set rate",
"IDC_RADIO_SCALE_WIDTH":"Set trans width",
Expand Down
3 changes: 2 additions & 1 deletion bin/lang/japanese.json
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{
{
"IDC_STATIC_IO_SETTING":"入出力設定",
"IDC_STATIC_INPUT_PATH":"入力パス\r\n(フォルダでも複数でもOK)",
"IDC_BUTTON_INPUT_REF":"参照",
Expand All @@ -20,6 +20,7 @@
"IDC_STATIC_JPEG_NOISE_LEVEL":"JPEGノイズ除去レベル",
"IDC_RADIONOISE_LEVEL1":"レベル1",
"IDC_RADIONOISE_LEVEL2":"レベル2",
"IDC_RADIONOISE_LEVEL3":"レベル3",
"IDC_STATIC_SCALE_RATE":"拡大サイズ",
"IDC_RADIO_SCALE_RATIO":"拡大率で指定",
"IDC_RADIO_SCALE_WIDTH":"変換後の横幅で指定",
Expand Down
3 changes: 2 additions & 1 deletion bin/lang/korean.json
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{
{
"IDC_STATIC_IO_SETTING":"입출력 설정",
"IDC_STATIC_INPUT_PATH":"입력 경로\r\n(파일 및 폴더)",
"IDC_BUTTON_INPUT_REF":"찾기",
Expand All @@ -20,6 +20,7 @@
"IDC_STATIC_JPEG_NOISE_LEVEL":"JPEG 노이즈 제거 단계",
"IDC_RADIONOISE_LEVEL1":"1단계",
"IDC_RADIONOISE_LEVEL2":"2단계",
"IDC_RADIONOISE_LEVEL3":"3단계",
"IDC_STATIC_SCALE_RATE":"확대 크기",
"IDC_RADIO_SCALE_RATIO":"확대율로 설정",
"IDC_RADIO_SCALE_WIDTH":"변환 후의 가로 너비",
Expand Down
3 changes: 2 additions & 1 deletion bin/lang/turkish.json
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
{
{
"IDC_STATIC_IO_SETTING":"Girdi/Çıktı Ayarları",
"IDC_STATIC_INPUT_PATH":"Girdi Yolu\r\n(dosya veya klasör)",
"IDC_BUTTON_INPUT_REF":"Gözat",
Expand All @@ -20,6 +20,7 @@
"IDC_STATIC_JPEG_NOISE_LEVEL":"JPEG gürültüsüzleştirme düzeyi",
"IDC_RADIONOISE_LEVEL1":"Seviye 1",
"IDC_RADIONOISE_LEVEL2":"Seviye 2",
"IDC_RADIONOISE_LEVEL3":"Seviye 3",
"IDC_STATIC_SCALE_RATE":"Büyütme oranı",
"IDC_RADIO_SCALE_RATIO":"Oran belirle",
"IDC_RADIO_SCALE_WIDTH":"Hedef en belirle",
Expand Down
1 change: 1 addition & 0 deletions bin/models/anime_style_art/noise3_model.json

Large diffs are not rendered by default.

Binary file not shown.
1 change: 1 addition & 0 deletions bin/models/anime_style_art_rgb/noise3_model.json

Large diffs are not rendered by default.

Binary file not shown.
1 change: 1 addition & 0 deletions bin/models/photo/noise3_model.json

Large diffs are not rendered by default.

Binary file added bin/models/photo/noise3_model.json.caffemodel
Binary file not shown.
Binary file modified waifu2x-caffe-gui/Resource.rc
Binary file not shown.
20 changes: 16 additions & 4 deletions waifu2x-caffe-gui/Source.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -280,8 +280,10 @@ class DialogEvent

if (SendMessage(GetDlgItem(dh, IDC_RADIONOISE_LEVEL1), BM_GETCHECK, 0, 0))
noise_level = 1;
else
else if (SendMessage(GetDlgItem(dh, IDC_RADIONOISE_LEVEL2), BM_GETCHECK, 0, 0))
noise_level = 2;
else
noise_level = 3;

if (SendMessage(GetDlgItem(dh, IDC_RADIO_SCALE_RATIO), BM_GETCHECK, 0, 0))
scaleType = eScaleTypeRatio;
Expand Down Expand Up @@ -1340,6 +1342,7 @@ class DialogEvent
SET_WINDOW_TEXT(IDC_STATIC_JPEG_NOISE_LEVEL);
SET_WINDOW_TEXT(IDC_RADIONOISE_LEVEL1);
SET_WINDOW_TEXT(IDC_RADIONOISE_LEVEL2);
SET_WINDOW_TEXT(IDC_RADIONOISE_LEVEL3);
SET_WINDOW_TEXT(IDC_STATIC_SCALE_RATE);
SET_WINDOW_TEXT(IDC_RADIO_SCALE_RATIO);
SET_WINDOW_TEXT(IDC_RADIO_SCALE_WIDTH);
Expand Down Expand Up @@ -1642,7 +1645,7 @@ class DialogEvent
if (outputExt.length() > 0 && outputExt[0] != TEXT('.'))
outputExt = L"." + outputExt;

if (!(1 <= noise_level && noise_level <= 2))
if (!(1 <= noise_level && noise_level <= 3))
noise_level = 1;

if (tprcess == TEXT("gpu"))
Expand Down Expand Up @@ -1717,13 +1720,21 @@ class DialogEvent
{
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL1), BM_SETCHECK, BST_CHECKED, 0);
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL2), BM_SETCHECK, BST_UNCHECKED, 0);
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL3), BM_SETCHECK, BST_UNCHECKED, 0);
}
else
else if (noise_level == 2)
{
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL1), BM_SETCHECK, BST_UNCHECKED, 0);
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL2), BM_SETCHECK, BST_CHECKED, 0);
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL3), BM_SETCHECK, BST_UNCHECKED, 0);
}

else
{
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL1), BM_SETCHECK, BST_UNCHECKED, 0);
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL2), BM_SETCHECK, BST_UNCHECKED, 0);
SendMessage(GetDlgItem(hWnd, IDC_RADIONOISE_LEVEL3), BM_SETCHECK, BST_CHECKED, 0);
}

if (process == "gpu")
{
SendMessage(GetDlgItem(hWnd, IDC_RADIO_MODE_GPU), BM_SETCHECK, BST_CHECKED, 0);
Expand Down Expand Up @@ -2286,6 +2297,7 @@ int WINAPI WinMain(HINSTANCE hInstance,
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIO_AUTO_SCALE);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIONOISE_LEVEL1);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIONOISE_LEVEL2);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIONOISE_LEVEL3);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIO_MODE_CPU);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIO_MODE_GPU);
cDialog.SetCommandCallBack(SetClassFunc(DialogEvent::UpdateAddString, &cDialogEvent), NULL, IDC_RADIO_MODEL_RGB);
Expand Down
Binary file modified waifu2x-caffe-gui/resource.h
Binary file not shown.
1 change: 1 addition & 0 deletions waifu2x-caffe/Source.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -93,6 +93,7 @@ int main(int argc, char** argv)
std::vector<int> cmdNRLConstraintV;
cmdNRLConstraintV.push_back(1);
cmdNRLConstraintV.push_back(2);
cmdNRLConstraintV.push_back(3);
TCLAP::ValuesConstraint<int> cmdNRLConstraint(cmdNRLConstraintV);
TCLAP::ValueArg<int> cmdNRLevel("n", "noise_level", "noise reduction level",
false, 1, &cmdNRLConstraint, cmd);
Expand Down

0 comments on commit 77b8497

Please sign in to comment.