임이지의 블로그

추론하려고 하는 모델의 인풋 속성이 INPUT 이미지에 속성이 224 x 224 x 3 크기로 사용했어야 하는데 모델 정보를 확인 안 하고 추론을 진행하였다. 오류 내용처럼 602112 바이트의 크기로 ByteBuffer를 만들어야 했었는 에 아래 코드는 224 x 224 * 1 의 크기로 ByteBuffer를 만들어서 전달하였다. private ByteBuffer convertBitmapToGrayByteBuffer(Bitmap bitmap) { ByteBuffer byteByffer = ByteBuffer.allocateDirect(bitmap.getByteCount()); byteByffer.order(ByteOrder.nativeOrder()); int[] pixels = new int[bitma..

AI 2022. 1. 10. 13:58

github.com/anishathalye/neural-style 상기 repositery에 tensorflow 소스를 빌드하는 도중 위와같은 오류가 발생하였다. tensorflow2.0, tenserflow 에서 나타는 에러이다. tensorflow에서는 ConfigProto() 또한 1.x 버전의 잔유물로 사용된다. tf.compat.v1.ConfigProto() 로 사용해야 한다. Session과 GPU를 잘 잡는지 확인하려고 보통 함께 사용하는 함수는 다음과 같다. sess = tf.Session(config=tf.ConfigProto(log_device_placement=True)) 상기에러를 수정하려면 아래와 같이 수정한다. sess = tf.compat.v1.Session(config=tf...

AI 2021. 4. 7. 17:34