如何解释Android上的mobilenetv2细分结果输出?

问题描述

我使用Deeplab's官方Github页面上提供的python脚本,用自己的数据集训练了量化语义细分模型。我使用了mobilenetv2_coco_voc_trainaug主干。我在Netron中检查了结果模型,并检查了输出的输入外观:

netron visualization

如您所见,输出是int64数组,大小为257x257。根据我的理解,此数组应包含每个数组索引中概率最高的label索引,还是我遗漏了什么? 但是,当我尝试在Android系统中阅读此书时,我得到的只是零和一,这与图片,人物,母牛等的含义无关。

@H_502_6@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) }

}

input image

segmentation result

结果也不是那么准确,因为我得到了不应该的细分值。 任何帮助将不胜感激!

解决方法

还不能发表评论,让我们尝试一下。

您正在尝试将量化模型与int64输出一起使用。输出应为8bit type

是的,量化模型的准确性将drop