init

APDrawingGAN2/data/single_dataset.py

@@ -136,11 +136,11 @@ class SingleDataset(BaseDataset):
                     xb = []
                     yb = []
                     for i in range(4):
-                        xbi = [int(center[i,0]-rws[i]/2), int(center[i,0]+rws[i]/2-1)]
-                        ybi = [int(center[i,1]-rhs[i]/2), int(center[i,1]+rhs[i]/2-1)]
+                        xbi = [center[i,0]-rws[i]/2, center[i,0]+rws[i]/2-1]
+                        ybi = [center[i,1]-rhs[i]/2, center[i,1]+rhs[i]/2-1]
                         for j in range(2):
-                            maskx = bound[:,xbi[j]]
-                            masky = bound[ybi[j],:]
+                            maskx = bound[:,int(xbi[j])]
+                            masky = bound[int(ybi[j]),:]
                             tmp_a = torch.from_numpy(maskx)*xbi[j].double()
                             tmp_b = torch.from_numpy(1-maskx)
                             xb += [tmp_b*10000 + tmp_a]