它的最后一层是输出为13x13x75矩阵的图层,由于输出类数为10,因此应为1x10。
它缺少层,包括层logit。下载模型时,它带有一些python文件供使用,并且在这些文件中,您看到了为获得1x10矩阵而必须完成的所有处理。
关键是该代码不应该存在,并且所有内容都必须在模型内。
扩展名为.pb的文件是什么?
有人知道会发生什么吗?
答案 0 :(得分:0)
很抱歉造成混乱。
扩展名为.pb的文件是什么? 那就是您加载以进行推理的TensorFlow模型。
根据收到的13 x 13 x 75的输出进行猜测,您创建了一个对象检测模型(OD),但是您期望获得1 x 10的输出,这向我表明您需要多标签模型。
Multi-Label:返回您期望的1x10向量。
OD:旨在提供边界框信息以及置信度;它需要自定义代码来计算管道中不支持的代码;这就是为什么需要在python代码('object_detection.py')中进行最终处理的原因。
kurt