a3d

Author: yi.he

alignshift/models/__init__.py

@@ -1,3 +1,3 @@
-from truncated_densenet3d_alignshift import DenseNetCustomTrunc3dAlign
-from truncated_densenet3d_tsm import DenseNetCustomTrunc3dTSM
-from truncated_densenet3d_a3d import DenseNetCustomTrunc3dA3D
+from .truncated_densenet3d_alignshift import DenseNetCustomTrunc3dAlign
+from .truncated_densenet3d_tsm import DenseNetCustomTrunc3dTSM
+from .truncated_densenet3d_a3d import DenseNetCustomTrunc3dA3D

deeplesion/ENVIRON.py

@@ -1 +1 @@
-data_root = '/mnt/data/DeepLesion/'
+data_root = '/mnt/data2/deeplesion/dataset/'

mmdet/datasets/__init__.py

@@ -12,10 +12,12 @@
 from deeplesion.dataset.DeepLesionDataset_tsm import DeepLesionDatasetTSM
 from deeplesion.dataset.DeepLesionDataset_align import DeepLesionDatasetAlign
 from deeplesion.dataset.DeepLesionDataset_25d import DeepLesionDataset25d
+from deeplesion.dataset.DeepLesionDataset_a3d import DeepLesionDatasetA3D
+
 __all__ = [
     'CustomDataset', 'XMLDataset', 'CocoDataset', 'VOCDataset',
     'CityscapesDataset', 'GroupSampler', 'DistributedGroupSampler',
     'build_dataloader', 'ConcatDataset', 'RepeatDataset', 'WIDERFaceDataset',
     'DATASETS', 'build_dataset', 'DeepLesionDatasetTSM',
-    'DeepLesionDatasetAlign', 'DeepLesionDataset25d'
+    'DeepLesionDatasetAlign', 'DeepLesionDataset25d', 'DeepLesionDatasetA3D'
 ]

mmdet/models/backbones/__init__.py

@@ -4,8 +4,8 @@
 from .ssd_vgg import SSDVGG
 
 from deeplesion.models.truncated_densenet import DenseNetCustomTrunc
-from deeplesion.models.truncated_densenet3d_tsm import DenseNetCustomTrunc3dTSM
-from deeplesion.models.truncated_densenet3d_a3d import DenseNetCustomTrunc3dA3D
-from deeplesion.models.truncated_densenet3d_alignshift import DenseNetCustomTrunc3dAlign
+from alignshift.models.truncated_densenet3d_tsm import DenseNetCustomTrunc3dTSM
+from alignshift.models.truncated_densenet3d_a3d import DenseNetCustomTrunc3dA3D
+from alignshift.models.truncated_densenet3d_alignshift import DenseNetCustomTrunc3dAlign
 __all__ = ['ResNet', 'make_res_layer', 'ResNeXt', 'SSDVGG', 'HRNet', 'DenseNetCustomTrunc3dTSM',
             'DenseNetCustomTrunc3dAlign', 'DenseNetCustomTrunc3dA3D', 'DenseNetCustomTrunc']

requirements.txt

@@ -1,4 +1,4 @@
-mmcv>=0.2.10
+mmcv==0.4.3
 numpy
 matplotlib
 six