From 92583036ad9dfff5e2db89f09c8ff08bd6dafbab Mon Sep 17 00:00:00 2001 From: "Ivan Kuchin (ikuchin)" Date: Mon, 8 Jul 2024 19:13:51 -0400 Subject: [PATCH] Code refactoring: move config to main folder --- tools/config.py => config.py | 0 dataset/craft_datasets.py | 2 +- dataset/ct150_preprocessing.py | 2 +- dataset/pomc_dataset.py | 2 +- predict.py | 2 +- tools/craft_network/__init__.py | 2 +- tools/craft_network/att_unet.py | 2 +- tools/craft_network/att_unet_dsv.py | 2 +- tools/craft_network/unet_classic.py | 2 +- tools/craft_network/unet_shortcuts_every_layer.py | 2 +- tools/evaluate.py | 2 +- tools/predict_on_random_data.py | 2 +- train_segmentation.py | 2 +- 13 files changed, 12 insertions(+), 12 deletions(-) rename tools/config.py => config.py (100%) diff --git a/tools/config.py b/config.py similarity index 100% rename from tools/config.py rename to config.py diff --git a/dataset/craft_datasets.py b/dataset/craft_datasets.py index 4d9e441..83f53f7 100644 --- 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 index b2935b5..52a8dcc 100644 --- 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 index 7c1f7df..ba49223 100644 --- 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 index d4d851e..0c65b7c 100644 --- 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 index a03be23..be15bd9 100644 --- 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 index 789f84f..3990cfe 100644 --- 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 index c2227e6..6d2a890 100644 --- 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 index 871329d..96e8f35 100644 --- 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 index b86f364..8841dad 100644 --- 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 index 0da2e2c..e1b04e3 100644 --- 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 index 7aea92f..f85eb12 100644 --- 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 index 009648d..aef4716 100644 --- 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():