gpt4 book ai didi

python - 在重新训练示例中使用 inception v4

转载 作者:太空狗 更新时间:2023-10-29 21:02:21 30 4
gpt4 key购买 nike

我正在尝试调整示例重新训练脚本 ( https://github.com/tensorflow/tensorflow/blob/master/tensorflow/examples/image_retraining/retrain.py ) 以使用 Inception V4 模型。

该脚本已经支持Inception V3 (2015) 以及不同版本的Mobilenets 的再训练。

到目前为止我做了什么:由于脚本使用 protobuf (.pb) 文件而不是检查点 (.ckpt),我从这里下载了 inception_v4.pb:https://deepdetect.com/models/tf/inception_v4.pb .据我所知,也可以加载检查点并使用卡住图工具来获取相同的文件。

然后,我使用 tensorflow python 工具 import_pb_to_tensorboard.py 查看了 tensorboard 中的图形,该工具可以在 tensorflow github 存储库中找到。从那里(如果我没记错请纠正我)我发现 resized_input_tensor_name 被称为 InputImagebottleneck_tensor_nameInceptionV4/Logits/Logits/MatMulbottleneck_tensor_size1001

有了这些信息,我尝试通过添加以下内容来调整重新训练脚本的 create_model_info(architecture) 函数:

    elif architecture == 'inception_v4':        data_url = 'http://download.tensorflow.org/models/image/imagenet/inception-2015-12-05.tgz' #this won't make any difference        bottleneck_tensor_name = 'InceptionV4/Logits/Logits/MatMul'        bottleneck_tensor_size = 1001        input_width = 299        input_height = 299        input_depth = 3        resized_input_tensor_name = 'InputImage'        model_file_name = 'inception_v4.pb'        input_mean = 128        input_std = 128

我使用以下命令运行脚本:

python retrain.py --architecture=inception_v4 --bottleneck_dir=test2/bottlenecks --model_dir=inception_v4 --summaries_dir=test2/summaries/basic --output_graph=test2/graph_flowers.pb --output_labels=test2/labels_flowers.txt --image_dir=datasets/flowers/flower_photos --how_many_training_steps 100

我收到以下错误:

File "retrain.py", line 373, in create_bottleneck_file str(e))) RuntimeError: Error during processing file datasets/flowers/flower_photos/tulips/4546299243_23cd58eb43.jpg (Cannot interpret feed_dict key as Tensor: Can not convert a Operation into a Tensor.)

最佳答案

我目前正在处理同样的事情。

尝试将 :0 添加到 bottleneck_tensor_nameresized_input_tensor_name 的末尾。

如果您在 retrain.py 中注意到, Google 也使用此 :0 命名法。

我的怀疑是,对于您来说,InceptionV4/Logits/Logits/MatMul 只是一个操作,您不想为此脚本获取它,而 InceptionV4/Logits/Logits/MatMul:0 是从该操作实例化的第一个张量,您正试图为此脚本获取它。

关于python - 在重新训练示例中使用 inception v4,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46829684/

30 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com