Browse Source

Fix LBPCriterion #316

nagadomi 5 năm trước cách đây
mục cha
commit
715b6f0d28
1 tập tin đã thay đổi với 2 bổ sung2 xóa
  1. 2 2
      lib/LBPCriterion.lua

+ 2 - 2
lib/LBPCriterion.lua

@@ -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[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
       end
       end
       model:add(filter)
       model:add(filter)