【发布时间】:2020-12-09 00:19:24
【问题描述】:
我使用Deeplab's 官方 Github 页面上提供的 python 脚本,使用我自己的数据集训练了一个量化语义分割模型。我使用了mobilenetv2_coco_voc_trainaug 主干。我检查了Netron 中的结果模型,这是输入输出的样子:
如您所见,输出是一个大小为 257x257 的 int64 数组。据我了解,该数组应包含每个数组索引处概率最高的标签索引,还是我遗漏了什么? 但是,当我尝试在 Android 中阅读此内容时,我得到的只是零和一,与图片、人物、牛等内容无关。
for (y in 0 until imageHeight) {
for (x in 0 until imageWidth) {
// resultBuffer is a ByteBuffer of size imageSize * imageSize * 8
val value = resultBuffer.getLong((y * imageWidth + x) * 8)
}
}
结果也不是那么准确,因为我得到了不应该的分段值。 任何帮助将不胜感激!
【问题讨论】:
标签: android tensorflow semantic-segmentation tensorflow-lite deeplab