Freeze.py 导出 pb 文件时，output_node_names 是什么

tfers-migration · April 1, 2020, 10:53am

我看了文档解释说 output_node_names 是模型的输出节点，我通过 python /tensorflow/python/tools/inspect_checkpoint.py --file_name=model-10000-ckpt

打印出了节点信息，但是我不知道怎么找到我要的那个节点，后来查资料说是在定义网络结构的时候，已经定义好了。

在这方面是小白，目前只会在网上在开源的训练，很多代码都看不懂，求大神帮忙解释下，这个模型里面的 SRGAN 网络结构的输出节点是什么啊？

github.com

brade31919/SRGAN-tensorflow/blob/master/lib/model.py

from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import tensorflow as tf
from lib.ops import *
import collections
import os
import math
import scipy.misc as sic
import numpy as np


# Define the dataloader
def data_loader(FLAGS):
    with tf.device('/cpu:0'):
        # Define the returned data batches
        Data = collections.namedtuple('Data', 'paths_LR, paths_HR, inputs, targets, image_count, steps_per_epoch')

        #Check the input directory

This file has been truncated. show original

chasehui 2018-7-12 21:57:19

tfers-migration · April 1, 2020, 10:53am

我觉得你可以试一下定义一下 name，这样子你打印 ckpt 里面的变量的时候，就好找了，，要不然像我用个 resnet50，然后变量多成狗，根本分不清哈哈哈

ViolinSolo 发表于 2018-7-13 15:41:42