add some feature

Author: REBOOTERS

subs/ai/build.gradle

@@ -37,6 +37,9 @@ android {
     androidResources {
         noCompress = 'tflite'
     }
+    buildFeatures {
+        viewBinding true
+    }
 
 }
 
@@ -55,4 +58,7 @@ dependencies {
         exclude group: "com.google.inject", module: "guice"
     }
     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'
 }

subs/ai/src/main/AndroidManifest.xml

@@ -5,6 +5,9 @@
         <activity
             android:name=".DigitalClassificationActivity"
             android:exported="true" />
+        <activity
+            android:name=".GanActivity"
+            android:exported="true" />
     </application>
 
 </manifest>

subs/ai/src/main/assets/dcgan.pt

@@ -0,0 +1,29 @@
+package com.engineer.ai
+
+import android.content.Context
+import java.io.File
+import java.io.FileOutputStream
+import java.io.IOException
+
+object AndroidAssetsFileUtil {
+
+
+    fun assetFilePath(context: Context, assetName: String): String {
+        val file = File(context.filesDir, assetName)
+        if (file.exists() && file.length() > 0) {
+            return file.absolutePath
+        }
+
+        return context.assets.open(assetName).use { inputStream ->
+            FileOutputStream(file).use { outputStream ->
+                val buffer = ByteArray(4 * 1024)
+                var read: Int
+                while (inputStream.read(buffer).also { read = it } != -1) {
+                    outputStream.write(buffer, 0, read)
+                }
+                outputStream.flush()
+            }
+            file.absolutePath
+        }
+    }
+}

subs/ai/src/main/java/com/engineer/ai/AndroidAssetsFileUtil.kt

@@ -0,0 +1,49 @@
+package com.engineer.ai
+
+import android.os.Bundle
+import androidx.appcompat.app.AppCompatActivity
+import com.engineer.ai.databinding.ActivityGanBinding
+import org.pytorch.IValue
+import org.pytorch.Module
+import org.pytorch.Tensor
+import java.util.Random
+
+
+class GanActivity : AppCompatActivity() {
+    private lateinit var module: Module
+    private val modelName = "dcgan.pt"
+
+    private lateinit var viewBinding: ActivityGanBinding
+    override fun onCreate(savedInstanceState: Bundle?) {
+        super.onCreate(savedInstanceState)
+        viewBinding = ActivityGanBinding.inflate(layoutInflater)
+        setContentView(viewBinding.root)
+        initModel()
+        viewBinding.gen.setOnClickListener {
+            genImage()
+        }
+    }
+
+    private fun genImage() {
+        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 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])
+
+    }
+
+    private fun initModel() {
+        module = Module.load(AndroidAssetsFileUtil.assetFilePath(this, modelName))
+    }
+
+
+}

subs/ai/src/main/java/com/engineer/ai/GanActivity.kt

@@ -0,0 +1,16 @@
+package com.engineer.ai
+
+import org.pytorch.Tensor
+
+class Mina {
+    fun kk() {
+        val zDim: IntArray = intArrayOf(1, 100)
+        val outDims: IntArray = intArrayOf(64, 64, 3)
+
+        val z = FloatArray(zDim[0] * outDims[1])
+
+        val shape = longArrayOf(1, 100)
+
+        val tensor = Tensor.fromBlob(z, shape)
+    }
+}

subs/ai/src/main/java/com/engineer/ai/Mina.kt

@@ -0,0 +1,22 @@
+<?xml version="1.0" encoding="utf-8"?>
+<androidx.constraintlayout.widget.ConstraintLayout xmlns:android="http://schemas.android.com/apk/res/android"
+    xmlns:app="http://schemas.android.com/apk/res-auto"
+    android:layout_width="match_parent"
+    android:layout_height="match_parent">
+
+
+    <Button
+        android:id="@+id/gen"
+        android:layout_width="match_parent"
+        android:layout_height="wrap_content"
+        android:layout_margin="10dp"
+        android:text="gan_gen"
+        app:layout_constraintBottom_toBottomOf="parent" />
+
+    <ImageView
+        android:id="@+id/gan_result"
+        android:layout_width="match_parent"
+        android:layout_height="wrap_content"
+        app:layout_constraintBottom_toTopOf="@id/gen" />
+
+</androidx.constraintlayout.widget.ConstraintLayout>