MarkWrobel commited on
Commit
acf1d8f
·
verified ·
1 Parent(s): 3d82b4a

Update utils/process_utils.py

Browse files
Files changed (1) hide show
  1. utils/process_utils.py +4 -4
utils/process_utils.py CHANGED
@@ -101,10 +101,10 @@ def apply_resp_model_28(input_im,frames = 32):
101
 
102
  #Applies super resolution model
103
  def apply_SR_model(input_im,frames = 32):
104
- E2E_model = os.path.join(MODELS_BASE, "E2E_SR_model")
105
- E2E = tf.keras.models.load_model(E2E_model, compile=False)
106
- weights = E2E.get_weights()
107
- sr_weights = weights[22:]
108
 
109
  inputs = tf.keras.Input(shape = [None,None,None,1])
110
  SR_model = build_3d_unet(input_shape=(None, None,None,1), num_classes=1)
 
101
 
102
  #Applies super resolution model
103
  def apply_SR_model(input_im,frames = 32):
104
+ E2E_model = os.path.join(MODELS_BASE, "E2E_SR_model")
105
+ E2E = tf.keras.models.load_model(E2E_model, compile=False)
106
+ weights = E2E.get_weights()
107
+ sr_weights = weights[22:]
108
 
109
  inputs = tf.keras.Input(shape = [None,None,None,1])
110
  SR_model = build_3d_unet(input_shape=(None, None,None,1), num_classes=1)