Mobilenet SSD输入形状
Mobilenet SSD输入形状
提问于 2019-12-26 11:43:41
我已经为单个对象检测任务在自定义数据集上训练了一个移动的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
Stack Overflow用户
回答已采纳
发布于 2020-01-02 07:00:49
形状: Batch_size，高度，宽度，通道
如果要更改该大小，则需要在转换为pb文件之前设置它，例如使用图表。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/59487979
复制相关文章
相似问题
腾讯云开发者
