1

我有一个来自 Tensorflow OD 的预训练 CocoModel，并在 kitti 数据集上对其进行了重新训练。

我只是想尝试对裁剪后的图像进行分类，但没有明确的文档说明张量的名称是什么。当我尝试使用 Tensorflow 的分类脚本时，它说 inceptionV3 模型中没有这样命名的张量。

有没有人已经尝试过并且知道张量的名称？

结果应该是可能的标签/类别及其分数的列表！

在此先感谢您的帮助

你好！感谢您的快速回复。

要获得概览：

  1. 我使用了 MSCOCO 的相关模型进行对象检测
  2. 我已经在具有 100 个图像的 KITTI 数据集上训练了模型
  3. 我已经导出了 inference_graph 并对其他图像进行了分类。
  4. 我为用户提供了一个标签工具来获取机器标签图像并更正它们。作为我想要实现的功能：

a) 绘制自己的边界框并标记它们（适用于 Canvas 和 Angular）

b）将此裁剪的边界框发送到我的服务器并从我的模型中获取类建议=>这是问题！我需要张量的名称来进行分类。

我试图重写自己的张量流分类图像.py https://github.com/tensorflow/models/blob/master/tutorials/image/imagenet/classify_image.py

4

1 回答 1

0

如果您的问题仅与张量名称有关，则此代码段可能会有所帮助，

ops = tf.get_default_graph().get_operations()
all_tensor_names = {output.name for op in ops for output in op.outputs}
tensor_dict = {}
for key in [
                'num_detections', 'detection_boxes', 'detection_scores',
                'detection_classes', 'detection_masks'
            ]:
tensor_name = key + ':0'
if tensor_name in all_tensor_names:
    tensor_dict[key] = tf.get_default_graph().get_tensor_by_name(
                        tensor_name)

此处的工作示例代码，https://github.com/dennywangtenk/balder/blob/master/Samples/test_pk_v1.py

于 2018-07-06T20:01:04.960 回答