fix: fix imports

Author: okunator

cellseg_models_pytorch/inference/post_processor.py

@@ -27,7 +27,6 @@
     "hovernet": True,
     "drfns": True,
     "dcan": False,
-    "dran": False,
 }
 
 ALLOWED_FORMATS = (".mat", ".geojson", ".feather", ".parquet")

cellseg_models_pytorch/losses/criterions/__init__.py

@@ -2,7 +2,6 @@
 from .ce import CELoss
 from .dice import DiceLoss
 from .focal import FocalLoss
-from .grad_mse import GradMSE
 from .iou import IoULoss
 from .mae import MAE
 from .mse import MSE
@@ -18,7 +17,6 @@
     "sce": SCELoss,
     "focal": FocalLoss,
     "mse": MSE,
-    "gmse": GradMSE,
     "ssim": SSIM,
     "msssim": MSSSIM,
     "mae": MAE,
@@ -30,7 +28,6 @@
     "SEG_LOSS_LOOKUP",
     "REG_LOSS_LOOKUP",
     "MSE",
-    "GradMSE",
     "SSIM",
     "MSSSIM",
     "DiceLoss",

cellseg_models_pytorch/postproc/__init__.py

@@ -9,7 +9,6 @@
     normalize_field,
 )
 from .functional.dcan import post_proc_dcan
-from .functional.dran import post_proc_dran
 from .functional.drfns import post_proc_drfns
 from .functional.hovernet import post_proc_hovernet
 from .functional.omnipose import get_masks_omnipose, post_proc_omnipose
@@ -24,7 +23,6 @@
     "omnipose": post_proc_omnipose,
     "dcan": post_proc_dcan,
     "drfns": post_proc_drfns,
-    "dran": post_proc_dran,
     "hovernet": post_proc_hovernet,
 }
 
@@ -44,6 +42,5 @@
     "post_proc_cellpose_old",
     "post_proc_drfns",
     "post_proc_dcan",
-    "post_proc_dran",
     "get_bboxes",
 ]

cellseg_models_pytorch/postproc/functional/stardist/nms.py

@@ -5,7 +5,7 @@
 from numba import njit, prange
 from scipy.spatial import KDTree
 
-from ....utils import intersection
+from cellseg_models_pytorch.utils import intersection
 
 __all__ = ["get_bboxes", "nms_stardist"]
 

cellseg_models_pytorch/utils/__init__.py

@@ -28,32 +28,9 @@
     type_map_flatten,
 )
 from .multiproc import run_pool
-from .tensor_kernels import filter2D, gaussian, gaussian_kernel2d, sobel_hv
-from .tensor_utils import (
-    ndarray_to_tensor,
-    tensor_one_hot,
-    tensor_to_ndarray,
-    to_device,
-)
-from .thresholding import (
-    argmax,
-    morph_chan_vese_thresh,
-    naive_thresh,
-    naive_thresh_prob,
-    niblack_thresh,
-    sauvola_thresh,
-)
-
-THRESH_LOOKUP = {
-    "argmax": argmax,
-    "naive": naive_thresh_prob,
-    "sauvola": sauvola_thresh,
-    "niblack": niblack_thresh,
-}
-
+from .tensor_utils import tensor_one_hot, to_device, to_tensor
 
 __all__ = [
-    "THRESH_LOOKUP",
     "Downloader",
     "FileHandler",
     "H5Handler",
@@ -75,18 +52,7 @@
     "remove_debris_semantic",
     "fill_holes_semantic",
     "label_semantic",
-    "naive_thresh_prob",
-    "naive_thresh",
-    "niblack_thresh",
-    "sauvola_thresh",
-    "morph_chan_vese_thresh",
-    "argmax",
-    "gaussian",
-    "gaussian_kernel2d",
-    "sobel_hv",
-    "filter2D",
-    "ndarray_to_tensor",
-    "tensor_to_ndarray",
+    "to_tensor",
     "to_device",
     "tensor_one_hot",
     "normalize_torch",