培训期间Tensorflow Slim调试

2024-05-18 14:22:07 发布

您现在位置:Python中文网/ 问答频道 /正文

对于初始模型,我主要遵循this。假设我要在每个训练循环上打印批处理中使用的图像数据,也就是说,我想打印中每个训练迭代的“images”变量的值 '在不同的标签集上微调模型。'代码,既然会话实际上是在苗条训练训练训练所以我不能做sess.运行([images])这里没有ses?在


Tags: 数据代码模型图像thissesssesimages
3条回答

使用以下脚本,您可以看到图像: summaries.add(tf.summary.image('image_tensor,max_images=15,name='images'))

tf.summary.image是最简单的方法,因为如前所述,会话op是在slim.learning.train中创建的。虽然interactivesession()可以在代码中间使用,但它可能会干扰slim创建的会话!在

这两个先例的答案都非常好地形象化的形象。在

但是,如果您有兴趣打印任何其他变量值甚至图像的像素强度,可以在调用之前使用以下代码苗条。学习。训练例如:

    variable = tf.get_default_graph().get_tensor_by_name("variable_name")
    train_tensor = tf.Print(train_tensor, [variable], text_to_print_before_value, summarize=number_of_values_to_print_if_array)

第一行通过其名称获取变量(您可以通过tensorboard或通过打印获得它)slim.get_model_变量()例如)。在

第二行在计算train_张量时打印它,并返回train_张量本身,以便将这个打印节点添加到图形中。在

从Mobilenet模型打印变量的示例:

^{pr2}$

所以你想把训练图像形象化?好吧,你可以在任何你有权限的地方添加一个图像摘要,例如在“Train the model on the Flowers dataset”下:

images, _, labels = load_batch(dataset)
tf.summary.image("input/images", images)

相关问题 更多 >

    热门问题