qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-08 11:20:48 +02:00
Code Issues Releases Wiki Activity

adjusting to tf2

Browse source
This commit is contained in:
vahidrezanezhad 2024-04-04 11:26:28 +02:00
parent dbb404030e
commit 522f00ab99
4 changed files with 15 additions and 23 deletions
Download patch file Download diff file Expand all files Collapse all files

4
build_model_load_pretrained_weights_and_save.py
View file

@ -1,8 +1,8 @@
import os import os
import sys import sys
import tensorflow as tf import tensorflow as tf
import keras , warnings import warnings
from keras.optimizers import * from tensorflow.keras.optimizers import *
from sacred import Experiment from sacred import Experiment
from models import * from models import *
from utils import * from utils import *

2
metrics.py
View file

@ -1,4 +1,4 @@
from keras import backend as K from tensorflow.keras import backend as K
import tensorflow as tf import tensorflow as tf
import numpy as np import numpy as np

8
models.py
View file

@ -1,7 +1,7 @@
from keras.models import * from tensorflow.keras.models import *
from keras.layers import * from tensorflow.keras.layers import *
from keras import layers from tensorflow.keras import layers
from keras.regularizers import l2 from tensorflow.keras.regularizers import l2
resnet50_Weights_path='./pretrained_model/resnet50_weights_tf_dim_ordering_tf_kernels_notop.h5' resnet50_Weights_path='./pretrained_model/resnet50_weights_tf_dim_ordering_tf_kernels_notop.h5'
IMAGE_ORDERING ='channels_last' IMAGE_ORDERING ='channels_last'

24
train.py
View file

@ -1,29 +1,21 @@
import os import os
import sys import sys
import tensorflow as tf import tensorflow as tf
from keras.backend.tensorflow_backend import set_session from tensorflow.compat.v1.keras.backend import set_session
import keras , warnings import warnings
from keras.optimizers import * from tensorflow.keras.optimizers import *
from sacred import Experiment from sacred import Experiment
from models import * from models import *
from utils import * from utils import *
from metrics import * from metrics import *
from keras.models import load_model from tensorflow.keras.models import load_model
from tqdm import tqdm from tqdm import tqdm
def configuration(): def configuration():
keras.backend.clear_session() config = tf.compat.v1.ConfigProto()
tf.reset_default_graph()
warnings.filterwarnings('ignore')
os.environ['CUDA_DEVICE_ORDER']='PCI_BUS_ID'
config = tf.ConfigProto(log_device_placement=False, allow_soft_placement=True)
config.gpu_options.allow_growth = True config.gpu_options.allow_growth = True
config.gpu_options.per_process_gpu_memory_fraction=0.95#0.95 session = tf.compat.v1.Session(config=config)
config.gpu_options.visible_device_list="0" set_session(session)
set_session(tf.Session(config=config))
def get_dirs_or_files(input_data): def get_dirs_or_files(input_data):
if os.path.isdir(input_data): if os.path.isdir(input_data):
@ -219,7 +211,7 @@ def run(n_classes,n_epochs,input_height,
validation_data=val_gen, validation_data=val_gen,
validation_steps=1, validation_steps=1,
epochs=1) epochs=1)
model.save(dir_output+'/'+'model_'+str(i)+'.h5') model.save(dir_output+'/'+'model_'+str(i))
#os.system('rm -rf '+dir_train_flowing) #os.system('rm -rf '+dir_train_flowing)