Merge pull request #27 from czw0078/master

fix the ghost problem and divided by zero sum problem
daniilidis-group · Oct 8, 2018 · 69506a8 · 69506a8
2 parents 534e206 + a2e45e0
commit 69506a8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/neural_renderer/cuda/load_textures_cuda_kernel.cu b/neural_renderer/cuda/load_textures_cuda_kernel.cu
@@ -41,7 +41,7 @@ __global__ void load_textures_cuda_kernel(
   scalar_t dim0 = ((i / (ts * ts)) % ts) / (ts - 1.) ;
   scalar_t dim1 = ((i / ts) % ts) / (ts - 1.);
   scalar_t dim2 = (i % ts) / (ts - 1.);
-  if (1 < dim0 + dim1 + dim2) {
+  if (0 < dim0 + dim1 + dim2) {
       float sum = dim0 + dim1 + dim2;
       dim0 /= sum;
       dim1 /= sum;