Skip to content

Commit

Permalink
Fix relative imports (#222)
Browse files Browse the repository at this point in the history
  • Loading branch information
vasudev-sharma authored and mathieuboudreau committed Jun 12, 2019
1 parent 804ca0b commit 09e3659
Show file tree
Hide file tree
Showing 7 changed files with 12 additions and 12 deletions.
6 changes: 3 additions & 3 deletions AxonDeepSeg/apply_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,9 +10,9 @@
import AxonDeepSeg.ads_utils
from AxonDeepSeg.ads_utils import convert_path

from .visualization.get_masks import get_masks
from .patch_management_tools import im2patches_overlap, patches2im_overlap
from .config_tools import update_config, default_configuration
from AxonDeepSeg.visualization.get_masks import get_masks
from AxonDeepSeg.patch_management_tools import im2patches_overlap, patches2im_overlap
from AxonDeepSeg.config_tools import update_config, default_configuration

def apply_convnet(path_acquisitions, acquisitions_resolutions, path_model_folder, config_dict, ckpt_name='model',
inference_batch_size=1, overlap_value=25, resampled_resolutions=[0.1],
Expand Down
2 changes: 1 addition & 1 deletion AxonDeepSeg/data_management/data_augmentation.py
Original file line number Diff line number Diff line change
Expand Up @@ -5,7 +5,7 @@
from skimage.filters import gaussian
import numpy as np
import random
from .patch_extraction import extract_patch
from AxonDeepSeg.data_management.patch_extraction import extract_patch
import AxonDeepSeg.ads_utils

#######################################################################################################################
Expand Down
2 changes: 1 addition & 1 deletion AxonDeepSeg/data_management/dataset_building.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,7 +7,7 @@

import AxonDeepSeg.ads_utils
from AxonDeepSeg.data_management.input_data import labellize_mask_2d
from .patch_extraction import extract_patch
from AxonDeepSeg.data_management.patch_extraction import extract_patch
from AxonDeepSeg.ads_utils import convert_path

def raw_img_to_patches(path_raw_data, path_patched_data, thresh_indices = [0, 0.2, 0.8],
Expand Down
2 changes: 1 addition & 1 deletion AxonDeepSeg/data_management/input_data.py
Original file line number Diff line number Diff line change
Expand Up @@ -14,7 +14,7 @@
# AxonDeepSeg imports
import AxonDeepSeg.ads_utils
from AxonDeepSeg.patch_management_tools import apply_legacy_preprocess, apply_preprocess
from .data_augmentation import *
from AxonDeepSeg.data_management.data_augmentation import *


def generate_list_transformations(transformations = {}, thresh_indices = [0,0.5], verbose=0):
Expand Down
6 changes: 3 additions & 3 deletions AxonDeepSeg/network_construction.py
Original file line number Diff line number Diff line change
Expand Up @@ -7,8 +7,8 @@
os.environ['TF_CPP_MIN_LOG_LEVEL']='2' # Get rid of Tensorflow warnings.
import tensorflow as tf
import pickle
from .data_management.input_data import input_data
from .config_tools import generate_config
from AxonDeepSeg.data_management.input_data import input_data
from AxonDeepSeg.config_tools import generate_config
import AxonDeepSeg.ads_utils


Expand Down Expand Up @@ -234,4 +234,4 @@ def uconv_net(x, training_config, phase, bn_updated_decay = None, verbose = True
final_result = tf.reshape(finalconv,
[tf.shape(finalconv)[0], data_temp_size[-1] * data_temp_size[-1], n_classes])

return final_result
return final_result
4 changes: 2 additions & 2 deletions AxonDeepSeg/train_network.py
Original file line number Diff line number Diff line change
Expand Up @@ -13,8 +13,8 @@
import AxonDeepSeg.ads_utils
from AxonDeepSeg.ads_utils import convert_path

from .data_management.input_data import input_data
from .config_tools import generate_config
from AxonDeepSeg.data_management.input_data import input_data
from AxonDeepSeg.config_tools import generate_config


def train_model(path_trainingset, path_model, config, path_model_init=None,
Expand Down
2 changes: 1 addition & 1 deletion AxonDeepSeg/visualization/visualize.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
# AxonDeepSeg imports
import AxonDeepSeg.ads_utils
from AxonDeepSeg.ads_utils import convert_path
from ..testing.segmentation_scoring import score_analysis, dice
from AxonDeepSeg.testing.segmentation_scoring import score_analysis, dice


def visualize_training(path_model, iteration_start_for_viz=0):
Expand Down

0 comments on commit 09e3659

Please sign in to comment.