fix: fix input size

greyovo · Oct 28, 2024 · 06c67ae · 06c67ae
1 parent def9345
commit 06c67ae
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 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
@@ -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
@@ -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
@@ -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> {