Mirror/waifu2x-caffe
1
0
Fork 0
mirror of https://github.com/lltcggie/waifu2x-caffe.git synced 2025-06-26 13:42:48 +00:00
Code Issues Packages Projects Releases Wiki Activity

block_sizeを変更出来るようにしてみた

Browse Source
This commit is contained in:
lltcggie 2015-06-03 00:12:40 +09:00
parent dde22d8e73
commit b6e104bed7
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
common/waifu2x.cpp
View File

@ -29,6 +29,10 @@ const auto offset = 0;
const auto layer_num = 7;
const auto output_size = block_size - offset * 2;
const auto block_width_height = block_size + layer_num * 2;
const auto original_width_height = 128 + layer_num * 2;
const int ConvertMode = CV_RGB2YUV;
const int ConvertInverseMode = CV_YUV2RGB;
@ -311,6 +315,14 @@ eWaifu2xError ConstractNet(boost::shared_ptr<caffe::Net<float>> &net, const std:
else
layer_param->mutable_relu_param()->set_engine(caffe::ReLUParameter_Engine_CAFFE);
}
else if (type == "MemoryData")
{
if (layer_param->mutable_memory_data_param()->width() == original_width_height && layer_param->mutable_memory_data_param()->height() == original_width_height)
{
layer_param->mutable_memory_data_param()->set_width(block_width_height);
layer_param->mutable_memory_data_param()->set_height(block_width_height);
}
}
}
net = boost::shared_ptr<caffe::Net<float>>(new caffe::Net<float>(param));