balancap / SSD-Tensorflow

Single Shot MultiBox Detector in TensorFlow
4.11k stars 1.89k forks source link

ValueError:"Can't load save_path when it is None." #358

Open yuye-img opened 5 years ago

yuye-img commented 5 years ago

from future import print_function

from hyperparams import Hyperparams as hp import numpy as np from data_load import load_data import tensorflow as tf from train import Graph from utils import load_spectrograms

def eval():

Load graph

g = Graph(mode="eval"); print("Evaluation Graph loaded")

# Load data
fpaths, text_lengths, texts = load_data(mode="eval")

# Parse
text = np.fromstring(texts[0], np.int32) # (None,)
fname, mel, mag = load_spectrograms(fpaths[0])

x = np.expand_dims(text, 0) # (1, None)
y = np.expand_dims(mel, 0) # (1, None, n_mels*r)
z = np.expand_dims(mag, 0) # (1, None, n_mfccs)

saver = tf.train.Saver()
with tf.Session() as sess:
    saver.restore(sess, tf.train.latest_checkpoint(hp.logdir)); print("Restored!")

    writer = tf.summary.FileWriter(hp.logdir,sess.graph)

    # Feed Forward
    ## mel
    y_hat = np.zeros((1, y.shape[1], y.shape[2]), np.float32)  # hp.n_mels*hp.r
    for j in range(y.shape[1]):
        _y_hat = sess.run(g.y_hat, {g.x: x, g.y: y_hat})
        y_hat[:, j, :] = _y_hat[:, j, :]

    ## mag
    merged, gs = sess.run([g.merged, g.global_step], {g.x:x, g.y:y, g.y_hat: y_hat, g.z: z})
    writer.add_summary(merged, global_step=gs)
    writer.close()

if name == 'main': eval() print("Done")

AlanHN commented 4 years ago

I meet the save error