CodeMirror/waifu2x
1
0
Fork 0
mirror of synced 2024-05-06 14:02:22 +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 if i == 1 then
-- channel identity -- channel identity
for j = 1, ch do for j = 1, ch do
filter.weight[i]:fill(0) filter.weight[j]:fill(0)
filter.weight[i][i][math.floor(k/2)+1][math.floor(k/2)+1] = 1 filter.weight[j][j][math.floor(k/2)+1][math.floor(k/2)+1] = 1
end end
end end
model:add(filter) model:add(filter)