CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-04-24 16:22:01 +12:00
Code Issues Releases Wiki Activity

Fix LBPCriterion #316

Browse source
This commit is contained in:
nagadomi 2019-11-27 13:52:28 +09:00
parent 950108d561
commit 715b6f0d28
1 changed files with 2 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

4
lib/LBPCriterion.lua
View file

@ -12,8 +12,8 @@ local function create_filters(ch, n, k, layers)
if i == 1 then
-- channel identity
for j = 1, ch do
filter.weight[i]:fill(0)
filter.weight[i][i][math.floor(k/2)+1][math.floor(k/2)+1] = 1
filter.weight[j]:fill(0)
filter.weight[j][j][math.floor(k/2)+1][math.floor(k/2)+1] = 1
end
end
model:add(filter)