我已经在自定义数据集上针对单个对象检测任务训练了移动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”)指的是什么?