init

APDrawingGAN2/data/single_dataset.py

@@ -136,8 +136,8 @@ class SingleDataset(BaseDataset):
                     xb = []
                     yb = []
                     for i in range(4):
-                        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]
+                        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)]
                         for j in range(2):
                             maskx = bound[:,xbi[j]]
                             masky = bound[ybi[j],:]