从Azure自定义视觉导出以使用TensorFlow的模型不完整

时间:2019-05-09 16:08:30

标签: azure tensorflow artificial-intelligence azure-cognitive-services microsoft-custom-vision

它的最后一层是输出为13x13x75矩阵的图层,由于输出类数为10,因此应为1x10。

它缺少层,包括层logit。下载模型时,它带有一些python文件供使用,并且在这些文件中,您看到了为获得1x10矩阵而必须完成的所有处理。

关键是该代码不应该存在,并且所有内容都必须在模型内。

扩展名为.pb的文件是什么?

有人知道会发生什么吗?

1 个答案:

答案 0 :(得分:0)

很抱歉造成混乱。

  
    

扩展名为.pb的文件是什么?     那就是您加载以进行推理的TensorFlow模型。

  

根据收到的13 x 13 x 75的输出进行猜测,您创建了一个对象检测模型(OD),但是您期望获得1 x 10的输出,这向我表明您需要多标签模型。

Multi-Label:返回您期望的1x10向量。

OD:旨在提供边界框信息以及置信度;它需要自定义代码来计算管道中不支持的代码;这就是为什么需要在python代码('object_detection.py')中进行最终处理的原因。

kurt