|
@@ -5,12 +5,8 @@ from keras.layers import Activation, Dropout, Flatten, Dense, BatchNormalization
|
|
|
from keras import backend as K
|
|
from keras import backend as K
|
|
|
import tensorflow as tf
|
|
import tensorflow as tf
|
|
|
|
|
|
|
|
-# trick to enable import of config
|
|
|
|
|
-import sys
|
|
|
|
|
-sys.path.insert(0, '..')
|
|
|
|
|
-
|
|
|
|
|
from . import metrics
|
|
from . import metrics
|
|
|
-from config import cnn_config as cfg
|
|
|
|
|
|
|
+from ..config import cnn_config as cfg
|
|
|
|
|
|
|
|
def generate_model_2D(_input_shape):
|
|
def generate_model_2D(_input_shape):
|
|
|
|
|
|
