Spaces:
Sleeping
Sleeping
Update utils/process_utils.py
Browse files- 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 |
-
|
| 105 |
-
|
| 106 |
-
|
| 107 |
-
|
| 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)
|