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

batch_sizeを2以上にすると強制終了するバグを修正

Browse Source
This commit is contained in:
lltcggie 2016-07-04 01:16:23 +09:00
parent 387c66891a
commit d11f22a98f
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
common/cNet.cpp
View File

@ -577,7 +577,7 @@ int cNet::GetInputMemorySize(const int crop_w, const int crop_h, const int outer
const int input_block_plane_size = input_block_width * input_block_height * mInputPlane;
return input_block_plane_size * sizeof(float);
return input_block_plane_size * batch_size * sizeof(float);
}
@ -592,7 +592,7 @@ int cNet::GetOutputMemorySize(const int crop_w, const int crop_h, const int oute
const int output_block_plane_size = output_block_width * output_block_height * mInputPlane;
return output_block_plane_size * sizeof(float);
return output_block_plane_size * batch_size * sizeof(float);
}
// ネットワークを使って画像を再構築する