@@ -29,12 +29,12 @@ def soft_erode(img: torch.Tensor) -> torch.Tensor: # type: ignore
2929 if len (img .shape ) == 4 :
3030 p1 = - (F .max_pool2d (- img , (3 , 1 ), (1 , 1 ), (1 , 0 )))
3131 p2 = - (F .max_pool2d (- img , (1 , 3 ), (1 , 1 ), (0 , 1 )))
32- return torch .min (p1 , p2 ) # type: ignore
32+ return torch .min (p1 , p2 )
3333 elif len (img .shape ) == 5 :
3434 p1 = - (F .max_pool3d (- img , (3 , 1 , 1 ), (1 , 1 , 1 ), (1 , 0 , 0 )))
3535 p2 = - (F .max_pool3d (- img , (1 , 3 , 1 ), (1 , 1 , 1 ), (0 , 1 , 0 )))
3636 p3 = - (F .max_pool3d (- img , (1 , 1 , 3 ), (1 , 1 , 1 ), (0 , 0 , 1 )))
37- return torch .min (torch .min (p1 , p2 ), p3 ) # type: ignore
37+ return torch .min (torch .min (p1 , p2 ), p3 )
3838
3939
4040def soft_dilate (img : torch .Tensor ) -> torch .Tensor : # type: ignore
@@ -48,9 +48,9 @@ def soft_dilate(img: torch.Tensor) -> torch.Tensor: # type: ignore
4848 https://github.com/jocpae/clDice/blob/master/cldice_loss/pytorch/soft_skeleton.py#L18
4949 """
5050 if len (img .shape ) == 4 :
51- return F .max_pool2d (img , (3 , 3 ), (1 , 1 ), (1 , 1 )) # type: ignore
51+ return F .max_pool2d (img , (3 , 3 ), (1 , 1 ), (1 , 1 ))
5252 elif len (img .shape ) == 5 :
53- return F .max_pool3d (img , (3 , 3 , 3 ), (1 , 1 , 1 ), (1 , 1 , 1 )) # type: ignore
53+ return F .max_pool3d (img , (3 , 3 , 3 ), (1 , 1 , 1 ), (1 , 1 , 1 ))
5454
5555
5656def soft_open (img : torch .Tensor ) -> torch .Tensor :
0 commit comments