upload phasenet
Browse files- phasenet/__pycache__/model.cpython-39.pyc +0 -0
- pipeline.py +2 -1
phasenet/__pycache__/model.cpython-39.pyc
CHANGED
|
Binary files a/phasenet/__pycache__/model.cpython-39.pyc and b/phasenet/__pycache__/model.cpython-39.pyc differ
|
|
|
pipeline.py
CHANGED
|
@@ -1,6 +1,7 @@
|
|
| 1 |
from typing import Dict, List
|
| 2 |
import numpy as np
|
| 3 |
import tensorflow as tf
|
|
|
|
| 4 |
|
| 5 |
from phasenet.model import ModelConfig, UNet
|
| 6 |
from phasenet.postprocess import extract_picks
|
|
@@ -27,7 +28,7 @@ class PreTrainedPipeline():
|
|
| 27 |
saver = tf.compat.v1.train.Saver(tf.compat.v1.global_variables())
|
| 28 |
init = tf.compat.v1.global_variables_initializer()
|
| 29 |
sess.run(init)
|
| 30 |
-
latest_check_point = tf.train.latest_checkpoint(
|
| 31 |
print(f"restoring model {latest_check_point}")
|
| 32 |
saver.restore(sess, latest_check_point)
|
| 33 |
|
|
|
|
| 1 |
from typing import Dict, List
|
| 2 |
import numpy as np
|
| 3 |
import tensorflow as tf
|
| 4 |
+
import os
|
| 5 |
|
| 6 |
from phasenet.model import ModelConfig, UNet
|
| 7 |
from phasenet.postprocess import extract_picks
|
|
|
|
| 28 |
saver = tf.compat.v1.train.Saver(tf.compat.v1.global_variables())
|
| 29 |
init = tf.compat.v1.global_variables_initializer()
|
| 30 |
sess.run(init)
|
| 31 |
+
latest_check_point = tf.train.latest_checkpoint(os.path.join(path, "model/190703-214543"))
|
| 32 |
print(f"restoring model {latest_check_point}")
|
| 33 |
saver.restore(sess, latest_check_point)
|
| 34 |
|