core

nndet/ptmodule/retinaunet/base.py

@@ -32,12 +32,12 @@ from nndet.utils.tensor import to_numpy
 from nndet.evaluator.det import BoxEvaluator
 from nndet.evaluator.seg import SegmentationEvaluator
 
-from nndet.detection.retina import BaseRetinaNet
-from nndet.detection.boxes.matcher import IoUMatcher
-from nndet.detection.boxes.sampler import HardNegativeSamplerBatched
-from nndet.detection.boxes.coder import CoderType, BoxCoderND
-from nndet.detection.boxes.anchors import get_anchor_generator
-from nndet.detection.boxes.utils import box_iou
+from nndet.core.retina import BaseRetinaNet
+from nndet.core.boxes.matcher import IoUMatcher
+from nndet.core.boxes.sampler import HardNegativeSamplerBatched
+from nndet.core.boxes.coder import CoderType, BoxCoderND
+from nndet.core.boxes.anchors import get_anchor_generator
+from nndet.core.boxes.utils import box_iou
 
 from nndet.ptmodule.base_module import LightningBaseModuleSWA, LightningBaseModule
 

nndet/ptmodule/retinaunet/v001.py

@@ -16,7 +16,7 @@ limitations under the License.
 
 from nndet.ptmodule.retinaunet.base import RetinaUNetModule
 
-from nndet.detection.boxes.matcher import ATSSMatcher
+from nndet.core.boxes.matcher import ATSSMatcher
 from nndet.models.heads.classifier import BCECLassifier
 from nndet.models.heads.regressor import GIoURegressor
 from nndet.models.heads.comb import DetectionHeadHNMNative

nndet/utils/analysis.py

@@ -37,7 +37,7 @@ from sklearn.metrics import confusion_matrix
 from torch import Tensor
 import SimpleITK as sitk
 
-from nndet.detection.boxes import box_iou_np, box_size_np
+from nndet.core.boxes import box_iou_np, box_size_np
 from nndet.io.load import load_pickle, save_json
 from nndet.utils.info import maybe_verbose_iterable
 

scripts/convert_seg2det.py

@@ -30,7 +30,7 @@ from scipy import ndimage
 from scipy.ndimage import label
 from tqdm import tqdm
 
-from nndet.detection.boxes import box_size_np
+from nndet.core.boxes import box_size_np
 from nndet.io import get_case_ids_from_dir, load_json, save_json
 from nndet.io.transforms.instances import get_bbox_np
 from nndet.utils.config import compose