Merge pull request #4 from IvanKuchin/development

Code refactoring: move config to main folder
IvanKuchin · Jul 8, 2024 · 4ddeb6f · 4ddeb6f
2 parents 8fd5142 + 9258303
commit 4ddeb6f
Show file tree

Hide file tree

Showing 13 changed files with 12 additions and 12 deletions.
diff --git a/tools/config.py → config.py b/tools/config.py → config.py
diff --git a/dataset/craft_datasets.py b/dataset/craft_datasets.py
@@ -11,7 +11,7 @@
 sys.path.insert(0, parentdir) 
 
 from tools import resize_3d
-import tools.config as config
+import config as config
 
 
 DEBUG_DATALOADER = False

diff --git a/dataset/ct150_preprocessing.py b/dataset/ct150_preprocessing.py
@@ -6,7 +6,7 @@
 import re
 import nibabel
 from tools import resize_3d
-import tools.config as config
+import config as config
 
 PATIENTS_SRC_FOLDER = "/docs/src/kt/datasets/ct-150/data/"
 LABELS_SRC_FOLDER = "/docs/src/kt/datasets/ct-150/labels/"

diff --git a/dataset/pomc_dataset.py b/dataset/pomc_dataset.py
@@ -14,7 +14,7 @@
 sys.path.insert(0, parentdir) 
 
 from tools import resize_3d
-import tools.config as config
+import config as config
 
 # INPUT_DIMS = np.array([config.IMAGE_DIMENSION_X, config.IMAGE_DIMENSION_Y, config.IMAGE_DIMENSION_Z])
 # AUGMENT_SCALE_FACTOR = 0.1

diff --git a/predict.py b/predict.py
@@ -6,7 +6,7 @@
 import numpy as np
 import nibabel as nib
 import tools.craft_network as craft_network
-import tools.config as config
+import config as config
 
 
 class Predict:

diff --git a/tools/craft_network/__init__.py b/tools/craft_network/__init__.py
@@ -3,7 +3,7 @@
 import tools.craft_network.att_unet
 import tools.craft_network.att_unet_dsv
 
-import tools.config as config
+import config as config
 
 
 def craft_network(weights_file):

diff --git a/tools/craft_network/att_unet.py b/tools/craft_network/att_unet.py
@@ -11,7 +11,7 @@
 from tools.predict_on_random_data import predict_on_random_data
 from tools.craft_network.att_gate import AttGate
 
-import tools.config as config
+import config as config
 
 
 def double_conv(filters, kernel_size=[3,3,1], apply_batchnorm=True, apply_dropout=False):

diff --git a/tools/craft_network/att_unet_dsv.py b/tools/craft_network/att_unet_dsv.py
@@ -12,7 +12,7 @@
 from tools.craft_network.att_gate import AttGate
 from tools.craft_network.dsv import DSV
 
-import tools.config as config
+import config as config
 
 
 def double_conv(filters, kernel_size=[3,3,1], apply_batchnorm=True, apply_dropout=False):

diff --git a/tools/craft_network/unet_classic.py b/tools/craft_network/unet_classic.py
@@ -3,7 +3,7 @@
 
 from tools.predict_on_random_data import predict_on_random_data
 
-import tools.config as config
+import config as config
 
 
 def double_conv(filters, kernel_size=[3,3,3], apply_batchnorm=True, apply_dropout=False):

diff --git a/tools/craft_network/unet_shortcuts_every_layer.py b/tools/craft_network/unet_shortcuts_every_layer.py
@@ -1,7 +1,7 @@
 import tensorflow as tf
 import os
 
-import tools.config as config
+import config as config
 from tools.craft_network.att_gate import AttGate
 
 

diff --git a/tools/evaluate.py b/tools/evaluate.py
@@ -3,7 +3,7 @@
 
 from dataset.craft_datasets import craft_datasets
 from tools.craft_network import craft_network
-import tools.config as config
+import config as config
 
 def __print_stat(data, title=""):
  print('-' * 100)

diff --git a/tools/predict_on_random_data.py b/tools/predict_on_random_data.py
@@ -1,6 +1,6 @@
 import time
 import tensorflow as tf
-import tools.config as config
+import config as config
 import numpy as np
 
 

diff --git a/train_segmentation.py b/train_segmentation.py
@@ -6,7 +6,7 @@
 from dataset.craft_datasets import craft_datasets, py_read_data_and_label, crop_to_shape
 from tools.categorical_metrics import CategoricalMetric, CategoricalF1, CustomCounter, CustomReduceMetric
 from tools.craft_network import craft_network
-import tools.config as config
+import config as config
 
 
 def get_tensorboard_log_dir():