Fix : "cannot convert between a tensorflow lite buffer with 602112 bytes and a bytebuffer with 150528 bytes"

[jeongiin]

오류 수정
위 깃헙을 많이 참고하여 오류를 수정했다.

오류 원인 result 배열과 이미지 bitmap 변환 시 float 배열을 사용하지 않았기 때문이었다.

1. convertBitmapToByteBuffer

imgData.putFloat(((value shr 16 and 0xFF)- IMAGE_MEAN )/ IMAGE_STD)
                imgData.putFloat(((value shr 8 and 0xFF) - IMAGE_MEAN)/ IMAGE_STD)
                imgData.putFloat(((value and 0xFF)- IMAGE_MEAN)/ IMAGE_STD)

2. Initialize

private var labelProb: Array<FloatArray>

3. SetKey.kt

const val IMAGE_MEAN: Int = 128 //0
const val IMAGE_STD: Float = 128.0f //0.toFloat

위 세 부분을 위와 같이 변경해주었더니 원하는 결과를 얻을 수 있었다.

참고 링크

kotlin Code 참고

기본 model 생성 Code 참고