Allow the output's channel number differs the input's channel number.

jimmy-ren · Apr 2, 2016 · 60321df · 60321df
1 parent 51e5428
commit 60321df
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/mem/conv2out_mem.m b/mem/conv2out_mem.m
@@ -8,8 +8,8 @@ function conv2out_mem()
     counts = ones(size(idx));
     counts = reshape(accumarray(idx(:), counts(:)), config.output_size(1), config.output_size(2));
     mem.one_over_add_counts = config.NEW_MEM(1 ./ counts);
-    mem.gen_out_matrix = config.NEW_MEM(zeros(size(idx, 1)*config.chs, size(idx, 2)));
-    for m = 1:config.chs
+    mem.gen_out_matrix = config.NEW_MEM(zeros(size(idx, 1)*config.output_size(3), size(idx, 2)));
+    for m = 1:config.output_size(3)%config.chs
         mem.gen_out_matrix((m-1)*size(idx, 1)+1:m*size(idx, 1), :) = idx + ((m-1)*max(max(idx)));
     end