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

batch_sizeを指定した場合、ネットに入力するデータのサイズが間違っていたのを優勢

Browse Source
This commit is contained in:
lltcggie 2015-06-07 04:52:35 +09:00
parent 0cb997f2b7
commit 4942a6d07b
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
common/waifu2x.cpp
View File

@ -455,7 +455,7 @@ Waifu2x::eWaifu2xError Waifu2x::ReconstructImage(boost::shared_ptr<caffe::Net<fl
} }
// ネットワークに画像を入力 // ネットワークに画像を入力
input_layer->Reset(block, dummy_data, input_block_plane_size * batch_size); input_layer->Reset(block, dummy_data, input_block_plane_size * processNum);
// 計算 // 計算
auto out = net->ForwardPrefilled(nullptr); auto out = net->ForwardPrefilled(nullptr);