qurator-spk/sbb_pixelwise_segmentation
1
0
Fork 0
mirror of https://github.com/qurator-spk/sbb_pixelwise_segmentation.git synced 2025-06-08 19:30:07 +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

8
models.py
View file

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