问题描述
我使用Deeplab's官方Github页面上提供的python脚本,用自己的数据集训练了量化语义细分模型。我使用了mobilenetv2_coco_voc_trainaug主干。我在Netron中检查了结果模型,并检查了输出的输入外观:
如您所见,输出是int64数组,大小为257x257。根据我的理解,此数组应包含每个数组索引中概率最高的label索引,还是我遗漏了什么? 但是,当我尝试在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)
}
}
结果也不是那么准确,因为我得到了不应该的细分值。 任何帮助将不胜感激!