fradinho commited on
Commit
d51c5c8
·
1 Parent(s): afdb9ed

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +6 -0
app.py CHANGED
@@ -9,6 +9,12 @@ from tensorflow.keras import backend as K
9
  from tensorflow.keras.models import Model
10
  import segmentation_models as sm
11
  from tensorflow import keras
 
 
 
 
 
 
12
 
13
  sm.set_framework('tf.keras')
14
 
 
9
  from tensorflow.keras.models import Model
10
  import segmentation_models as sm
11
  from tensorflow import keras
12
+ BACKBONE = 'efficientnetb7'
13
+ model10 = sm.FPN(BACKBONE,
14
+ #encoder_freeze = True,
15
+ #pyramid_block_filters=16,
16
+ encoder_weights='imagenet',
17
+ classes=n_classes, activation='softmax')
18
 
19
  sm.set_framework('tf.keras')
20