add pytorch dcgan

Author: REBOOTERS

Diff for: build.gradle

@@ -1,8 +1,8 @@
 // Top-level build file where you can add configuration options common to all sub-projects/modules.
 
 plugins {
-    id 'com.android.application' version '8.8.0' apply false
-    id 'com.android.library' version '8.8.0' apply false
+    id 'com.android.application' version '8.8.1' apply false
+    id 'com.android.library' version '8.8.1' apply false
     id 'org.jetbrains.kotlin.android' version '2.0.20' apply false
     id 'com.google.devtools.ksp' version '2.0.10-1.0.24' apply false
     id 'org.jetbrains.kotlin.plugin.compose' version '2.0.20' apply false

Diff for: buildSrc/build.gradle

@@ -9,7 +9,7 @@ dependencies {
 //    implementation localGroovy()
     implementation 'org.jetbrains.kotlin:kotlin-stdlib:2.1.0'
     // 添加了这个，就可以看 Android Gradle 插件的源码了
-    implementation 'com.android.tools.build:gradle-api:8.8.0'
+    implementation 'com.android.tools.build:gradle-api:8.8.1'
 //    implementation "org.jetbrains.kotlin:kotlin-script-runtime:1.3.40"
     implementation 'com.google.code.gson:gson:2.11.0'
     implementation 'com.android.tools:common:31.7.3'

Diff for: imitate/src/main/java/com/engineer/imitate/ui/fragments/EntranceFragment.kt

@@ -20,7 +20,7 @@ import androidx.recyclerview.widget.RecyclerView
 import com.alibaba.android.arouter.facade.annotation.Route
 import com.andrefrsousa.superbottomsheet.SuperBottomSheetFragment
 import com.bumptech.glide.Glide
-import com.engineer.ai.DigitalClassificationActivity
+import com.engineer.ai.AIHomeActivity
 import com.engineer.imitate.R
 import com.engineer.imitate.databinding.FragmentEntranceBinding
 import com.engineer.imitate.ui.activity.CLActivity
@@ -89,7 +89,7 @@ class EntranceFragment : Fragment() {
         }
 
         viewBinding.ai.setOnClickListener {
-            startActivity(Intent(context, DigitalClassificationActivity::class.java))
+            startActivity(Intent(context, AIHomeActivity::class.java))
         }
 
         viewBinding.scanWifi.setOnClickListener {

Diff for: subs/ai/build.gradle

@@ -59,6 +59,9 @@ dependencies {
     }
     implementation 'com.google.android.gms:play-services-tasks:18.2.0'
 
-    implementation 'org.pytorch:pytorch_android_lite:1.9.0'
-    implementation 'org.pytorch:pytorch_android_torchvision:1.9.0'
+
+    implementation 'org.pytorch:pytorch_android_lite:1.12.2'
+    implementation 'org.pytorch:pytorch_android_torchvision_lite:1.12.2'
+//    implementation 'org.pytorch:pytorch_android:1.10.0'
+//    implementation 'org.pytorch:pytorch_android_torchvision:1.10.0'
 }

Diff for: subs/ai/src/main/AndroidManifest.xml

@@ -2,6 +2,7 @@
 <manifest xmlns:android="http://schemas.android.com/apk/res/android">
 
     <application>
+        <activity android:name=".AIHomeActivity" android:exported="true" />
         <activity
             android:name=".DigitalClassificationActivity"
             android:exported="true" />

Diff for: subs/ai/src/main/assets/dcgan.pt

@@ -0,0 +1,32 @@
+package com.engineer.ai
+
+import android.content.Intent
+import android.os.Bundle
+import android.widget.Button
+import android.widget.LinearLayout
+import androidx.appcompat.app.AppCompatActivity
+
+
+class AIHomeActivity : AppCompatActivity() {
+
+    private val pages = arrayOf(GanActivity::class.java,
+        DigitalClassificationActivity::class.java)
+
+    override fun onCreate(savedInstanceState: Bundle?) {
+        super.onCreate(savedInstanceState)
+        val linerLayout = LinearLayout(this)
+        linerLayout.orientation = LinearLayout.VERTICAL
+
+        for (page in pages) {
+            val button = Button(this)
+            button.text = page.simpleName
+            button.setOnClickListener {
+                startActivity(Intent(this, page))
+            }
+            linerLayout.addView(button)
+        }
+        setContentView(linerLayout)
+    }
+
+
+}

Diff for: subs/ai/src/main/java/com/engineer/ai/AIHomeActivity.kt

@@ -9,6 +9,7 @@ import android.view.MotionEvent
 import android.widget.Button
 import android.widget.TextView
 import com.divyanshu.draw.widget.DrawView
+import com.engineer.ai.util.DigitClassifier
 import org.tensorflow.lite.TensorFlowLite
 
 class DigitalClassificationActivity : AppCompatActivity() {

Diff for: subs/ai/src/main/java/com/engineer/ai/DigitalClassificationActivity.kt

@@ -3,7 +3,10 @@ package com.engineer.ai
 import android.os.Bundle
 import androidx.appcompat.app.AppCompatActivity
 import com.engineer.ai.databinding.ActivityGanBinding
+import com.engineer.ai.util.AndroidAssetsFileUtil
+import com.engineer.ai.util.Utils
 import org.pytorch.IValue
+import org.pytorch.LiteModuleLoader
 import org.pytorch.Module
 import org.pytorch.Tensor
 import java.util.Random
@@ -28,21 +31,35 @@ class GanActivity : AppCompatActivity() {
         val zDim = intArrayOf(1, 100)
         val outDims = intArrayOf(64, 64, 3)
 
-        val z = FloatArray(zDim[0] * outDims[1])
-        val random = Random(System.currentTimeMillis())
-        z[0] = random.nextGaussian().toFloat()
+        val z = FloatArray(zDim[0] * zDim[1])
+
+        val rand = Random()
+        // 生成高斯随机数
+        for (c in 0 until zDim[0] * zDim[1]) {
+            z[c] = rand.nextGaussian().toFloat()
+        }
         val shape = longArrayOf(1, 100)
         val tensor = Tensor.fromBlob(z, shape)
 
         val resultT = module.forward(IValue.from(tensor)).toTensor()
         val resultArray = resultT.dataAsFloatArray
-
-//        val img = floatArrayOf(outDims[0],outDims[1],outDims[2])
-
+        val resultImg = Array(outDims[0]) { Array(outDims[1]) { FloatArray(outDims[2]) { 0.0f } } }
+        var index = 0
+        // 根据输出的一维数组，解析生成的卡通图像
+        for (j in 0 until outDims[2]) {
+            for (k in 0 until outDims[0]) {
+                for (m in 0 until outDims[1]) {
+                    resultImg[k][m][j] = resultArray[index] * 127.5f + 127.5f
+                    index++
+                }
+            }
+        }
+        val bitmap = Utils.getBitmap(resultImg, outDims)
+        viewBinding.ganResult.setImageBitmap(bitmap)
     }
 
     private fun initModel() {
-        module = Module.load(AndroidAssetsFileUtil.assetFilePath(this, modelName))
+        module = LiteModuleLoader.load(AndroidAssetsFileUtil.assetFilePath(this, modelName))
     }
 
 

Diff for: subs/ai/src/main/java/com/engineer/ai/GanActivity.kt

@@ -1,9 +1,8 @@
-package com.engineer.ai
+package com.engineer.ai.util
 
 import android.content.Context
 import java.io.File
 import java.io.FileOutputStream
-import java.io.IOException
 
 object AndroidAssetsFileUtil {
 

Diff for: subs/ai/src/main/java/com/engineer/ai/Mina.kt

@@ -11,7 +11,7 @@ See the License for the specific language governing permissions and
 limitations under the License.
 ==============================================================================*/
 
-package com.engineer.ai
+package com.engineer.ai.util
 
 import android.content.Context
 import android.content.res.AssetManager

Diff for: subs/ai/src/main/java/com/engineer/ai/AndroidAssetsFileUtil.kt renamed to subs/ai/src/main/java/com/engineer/ai/util/AndroidAssetsFileUtil.kt

@@ -0,0 +1,114 @@
+package com.engineer.ai.util;
+
+import android.content.Context;
+import android.database.Cursor;
+import android.graphics.Bitmap;
+import android.graphics.BitmapFactory;
+import android.graphics.Matrix;
+import android.net.Uri;
+import android.provider.MediaStore;
+
+public class Utils {
+
+    /**
+     * @param context 上下文
+     * @param uri 资源标识
+     * @return 路径值
+     */
+    public static String getPathFromUri(Context context, Uri uri) {
+
+        String result;
+        Cursor cursor = context.getContentResolver().query(uri, null, null, null, null);
+        if (cursor == null) {
+            result = uri.getPath();
+        } else {
+            cursor.moveToFirst();
+            int idx = cursor.getColumnIndex(MediaStore.Images.ImageColumns.DATA);
+            result = cursor.getString(idx);
+            cursor.close();
+        }
+        return result;
+    }
+
+    /**
+     * @param image_array 输入的四维数组
+     * @param dim_info 未读信息
+     * @return
+     */
+
+    public static Bitmap getBitmap(float[][][] image_array, int[] dim_info) {
+        int count = 0;
+        int[] color_info = new int[dim_info[0] * dim_info[1]];
+        // 遍历图像,获取颜色信息
+        for (int i = 0; i < dim_info[0]; i++) {
+            for (int j = 0; j < dim_info[1]; j++) {
+                float[] arr = image_array[i][j];
+                int alpha = 255;
+                int red = (int) arr[0];
+                int green = (int) arr[1];
+                int blue = (int) arr[2];
+                int tempARGB = (alpha << 24) | (red << 16) | (green << 8) | blue;
+                color_info[count++] = tempARGB;
+            }
+        }
+        // 创建bitmap对象
+        return Bitmap.createBitmap(color_info, dim_info[0], dim_info[1], Bitmap.Config.ARGB_8888);
+    }
+
+    /**
+     * @param filePath 文件路径
+     * @return Bitmap对象
+     */
+    public static Bitmap getScaleBitmapByPath(String filePath) {
+
+        BitmapFactory.Options options = new BitmapFactory.Options();
+        options.inJustDecodeBounds = true;
+        BitmapFactory.decodeFile(filePath, options);
+        int width = options.outWidth;
+        int height = options.outHeight;
+
+        int maxSize = 500;
+        options.inSampleSize = 1;
+        while (true) {
+            if (width / options.inSampleSize < maxSize || height / options.inSampleSize < maxSize) {
+                break;
+            }
+            options.inSampleSize *= 2;
+        }
+        options.inJustDecodeBounds = false;
+
+        // 返回解码后的图片
+        return BitmapFactory.decodeFile(filePath, options);
+    }
+
+
+    /**
+     * @param origin 原始Bitmap
+     * @param newWidth 缩放后的宽度
+     * @param newHeight 缩放后的高度
+     * @return 缩放后的Bitmap
+     */
+    public static Bitmap getScaleBitmapByBitmap(Bitmap origin, int newWidth, int newHeight) {
+        // 如果输入的Bitmap为空，则直接返回
+        if (origin == null) {
+            return null;
+        }
+        // 原始Bitmap的长宽
+        int height = origin.getHeight();
+        int width = origin.getWidth();
+
+        // 计算缩放后的图像比例
+        float scaleWidthRatio= ((float) newWidth) / width;
+        float scaleHeightRatio = ((float) newHeight) / height;
+
+        Matrix matrix = new Matrix();
+        matrix.postScale(scaleWidthRatio, scaleHeightRatio);
+
+        // 创建新的Bitmap
+        Bitmap scaledBitmap = Bitmap.createBitmap(origin, 0, 0, width, height, matrix, false);
+        if (!origin.isRecycled()) {
+            origin.recycle();
+        }
+        return scaledBitmap;
+    }
+}