From 06c67aeb12b98ac66ce3be2c8538df557bb5d486 Mon Sep 17 00:00:00 2001
From: seczhou <jelychow@gmail.com>
Date: Mon, 28 Oct 2024 19:22:49 +0800
Subject: [PATCH] fix: fix input size

---
 .../main/java/me/grey/picquery/feature/clip/ImageEncoderCLIP.kt | 2 +-
 .../picquery/feature/mobileclip2/ImageEncoderMobileCLIPv2.kt    | 2 +-
 .../picquery/feature/mobileclip2/PreprocessorMobileCLIPv2.kt    | 2 +-
 3 files changed, 3 insertions(+), 3 deletions(-)

diff --git a/app/src/main/java/me/grey/picquery/feature/clip/ImageEncoderCLIP.kt b/app/src/main/java/me/grey/picquery/feature/clip/ImageEncoderCLIP.kt
index bd748fd..ca60cbb 100644
--- a/app/src/main/java/me/grey/picquery/feature/clip/ImageEncoderCLIP.kt
+++ b/app/src/main/java/me/grey/picquery/feature/clip/ImageEncoderCLIP.kt
@@ -38,7 +38,7 @@ class ImageEncoderCLIP(context: Context, private val preprocessor: PreprocessorC
         }
     }
 
-    fun splitFloatBuffer(buffer: FloatBuffer, parts: Int): List<FloatBuffer> {
+    private fun splitFloatBuffer(buffer: FloatBuffer, parts: Int): List<FloatBuffer> {
         val totalSize = buffer.capacity()
         val partSize = totalSize / parts
         val result = mutableListOf<FloatBuffer>()
diff --git a/app/src/main/java/me/grey/picquery/feature/mobileclip2/ImageEncoderMobileCLIPv2.kt b/app/src/main/java/me/grey/picquery/feature/mobileclip2/ImageEncoderMobileCLIPv2.kt
index 9543380..1d1c660 100644
--- a/app/src/main/java/me/grey/picquery/feature/mobileclip2/ImageEncoderMobileCLIPv2.kt
+++ b/app/src/main/java/me/grey/picquery/feature/mobileclip2/ImageEncoderMobileCLIPv2.kt
@@ -25,7 +25,7 @@ class ImageEncoderMobileCLIPv2(context: Context, private val preprocessor: Prepr
 
     companion object {
         private const val TAG = "ImageEncoderLiteRT"
-        private const val IMAGE_SIZE = 224
+        private const val IMAGE_SIZE = 256
         private const val MODEL_PATH = "mobileclip-image.tflite"
     }
 
diff --git a/app/src/main/java/me/grey/picquery/feature/mobileclip2/PreprocessorMobileCLIPv2.kt b/app/src/main/java/me/grey/picquery/feature/mobileclip2/PreprocessorMobileCLIPv2.kt
index e96fa0b..5afb7a7 100644
--- a/app/src/main/java/me/grey/picquery/feature/mobileclip2/PreprocessorMobileCLIPv2.kt
+++ b/app/src/main/java/me/grey/picquery/feature/mobileclip2/PreprocessorMobileCLIPv2.kt
@@ -14,7 +14,7 @@ class PreprocessorMobileCLIPv2 : Preprocessor {
 
     companion object {
         // FIXME 应为 256，但输入模型时出错，可能是模型转换有问题。
-        const val IMAGE_SIZE = 224
+        const val IMAGE_SIZE = 256
     }
 
     override suspend fun preprocessBatch(input: List<Bitmap>): List<TensorBuffer> {