用inceptionV3 Mod从tensorflow中得到一个Cropped图像的分类

2024-04-20 15:04:11 发布

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

我有一个来自Tensorflow OD的预训练协整模型,并在kitti数据集上重新训练它。你知道吗

我只是想尝试对一张皱巴巴的图片进行分类,但是没有明确的文档记录这些张量的名称。当我尝试使用Tensorflow的classify脚本时,它说在inceptionV3模型中没有这样命名的张量。你知道吗

有没有人已经试过了,知道张量的名字?你知道吗

结果应该是一个可能的标签/类及其分数的列表!你知道吗

提前谢谢你的帮助

嗨! 谢谢你的快速回复。你知道吗

要获得概述:

  1. 我使用了MSCOCO的相关模型进行目标检测
  2. 我在一个有100张图片的KITTI数据集上训练了这个模型
  3. 我已经导出了推理图并对其他图像进行了分类。你知道吗
  4. 我为用户提供了一个标签工具来获取机器标签图像并进行校正。作为我想要实现的功能:

a)绘制自己的边界框并贴上标签(适用于画布和角度)

b)将此裁剪的边界框发送到我的服务器并从我的模型中获取类建议=>;问题出在这里!!!我需要张量的名字来分类。你知道吗

我试着重写自己的张量流_图片.pyhttps://github.com/tensorflow/models/blob/master/tutorials/image/imagenet/classify_image.py


Tags: 数据文档模型图像imagetensorflow分类图片
1条回答
网友
1楼 · 发布于 2024-04-20 15:04:11

如果你的问题只是关于张量的名字,这个片段会有帮助

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

相关问题 更多 >