Mobilenet SSD输入形状

时间:2019-12-26 11:43:41

标签: numpy deep-learning object-detection mobilenet tf-lite

我已经在自定义数据集上针对单个对象检测任务训练了移动SSD V2模型,并将其转换为tflite。当我使用解释程序加载.tflite模型进行测试,并使用input_details = model.get_input_details()获取输入详细信息时,它将输出

[{'name': 'normalized_input_image_tensor',
'index': 272,
'shape': array([  1, 300, 300,   3], dtype=int32),
'dtype': numpy.uint8,
'quantization': (0.0078125, 128)}]

我知道“ 300x300”是图像的高度和宽度,“ 3”是RGB通道,但是第一个元素(“ 1”)指的是什么?

1 个答案:

答案 0 :(得分:0)

形状:[批量大小,高度,宽度,通道]

如果要更改该大小,则需要先进行设置,然后再转换为pb文件,例如使用export_inference_graph