Update sam2point/utils.py

sam2point/utils.py

@@ -8,8 +8,6 @@ import torch
 def build_fold(path):
     if os.path.exists(path):
         return True
-        # shutil.rmtree(path)
-        # return True 
     os.makedirs(path)
     return False
     
@@ -38,11 +36,8 @@ def visualize_frame_with_mask(grid0, grid1, grid2, a0_mask, a1_mask, a2_mask, po
 
 def visualize_per_frame(grid, foldpath, resolution, args=None):
     grid = torch.nn.functional.interpolate(grid, size=(resolution, resolution), mode=args.mode)
-    # gridb = torch.nn.functional.interpolate(grid, size=(256, 256), mode='nearest')
-    # grid = grida * 0.8 + gridb * 0.2
     
     imgs = grid.cpu().numpy()
-    #print(imgs[0, :, 0:3, 0:3])
     n, _, _, _ = grid.shape
     for ii in range(n):
         r = np.uint8(imgs[ii, 0, :, :]*255)
@@ -50,12 +45,6 @@ def visualize_per_frame(grid, foldpath, resolution, args=None):
         b = np.uint8(imgs[ii, 2, :, :]*255)
         img = cv2.merge([b, g, r])
         
-        # bilateralFilter
-        # img = cv2.bilateralFilter(img, d=-1, sigmaColor=25, sigmaSpace=7)
-        # img = cv2.bilateralFilter(img, d=9, sigmaColor=50, sigmaSpace=16)
-        # img = cv2.GaussianBlur(img, (5, 5), 0)
-        # img = cv2.medianBlur(img, 5)
-
         cv2.imwrite('{}/{}.png'.format(foldpath, ii), img)
     return