Skip to content

Commit e3f91de

Browse files
committed
update ai module
1 parent c23efca commit e3f91de

File tree

3 files changed

+23
-10
lines changed

3 files changed

+23
-10
lines changed

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

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
<application>
55
<activity
66
android:name=".DigitalClassificationActivity"
7-
android:exported="false" />
7+
android:exported="true" />
88
</application>
99

1010
</manifest>

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

+8-3
Original file line numberDiff line numberDiff line change
@@ -14,6 +14,7 @@ import org.tensorflow.lite.TensorFlowLite
1414
class DigitalClassificationActivity : AppCompatActivity() {
1515
private var drawView: DrawView? = null
1616
private var clearButton: Button? = null
17+
private var initButton: Button? = null
1718
private var predictedTextView: TextView? = null
1819
private var digitClassifier = DigitClassifier(this)
1920

@@ -28,6 +29,7 @@ class DigitalClassificationActivity : AppCompatActivity() {
2829
drawView?.setColor(Color.WHITE)
2930
drawView?.setBackgroundColor(Color.BLACK)
3031
clearButton = findViewById(R.id.clear_button)
32+
initButton = findViewById(R.id.init_model)
3133
predictedTextView = findViewById(R.id.predicted_text)
3234

3335
// Setup clear drawing button.
@@ -52,8 +54,11 @@ class DigitalClassificationActivity : AppCompatActivity() {
5254
Log.d(TAG,"ver ${TensorFlowLite.schemaVersion()}")
5355
Log.d(TAG,"ver ${TensorFlowLite.runtimeVersion()}")
5456
// Setup digit classifier.
55-
digitClassifier.initialize()
56-
.addOnFailureListener { e -> Log.e(TAG, "Error to setting up digit classifier.", e) }
57+
initButton?.setOnClickListener {
58+
digitClassifier.initialize()
59+
.addOnFailureListener { e -> Log.e(TAG, "Error to setting up digit classifier.", e) }
60+
}
61+
5762
}
5863

5964
override fun onDestroy() {
@@ -78,6 +83,6 @@ class DigitalClassificationActivity : AppCompatActivity() {
7883
}
7984

8085
companion object {
81-
private const val TAG = "MainActivity"
86+
private const val TAG = "DigitalClassification"
8287
}
8388
}
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,5 @@
11
<?xml version="1.0" encoding="utf-8"?>
2-
<androidx.constraintlayout.widget.ConstraintLayout
3-
xmlns:android="http://schemas.android.com/apk/res/android"
2+
<androidx.constraintlayout.widget.ConstraintLayout xmlns:android="http://schemas.android.com/apk/res/android"
43
xmlns:app="http://schemas.android.com/apk/res-auto"
54
xmlns:tools="http://schemas.android.com/tools"
65
android:layout_width="match_parent"
@@ -12,19 +11,19 @@
1211
android:layout_width="match_parent"
1312
android:layout_height="0dp"
1413
app:layout_constraintDimensionRatio="1:1"
15-
app:layout_constraintTop_toTopOf="parent"/>
14+
app:layout_constraintTop_toTopOf="parent" />
1615

1716
<TextView
1817
android:id="@+id/predicted_text"
19-
android:textStyle="bold"
2018
android:layout_width="wrap_content"
2119
android:layout_height="wrap_content"
2220
android:text="@string/prediction_text_placeholder"
2321
android:textSize="20sp"
22+
android:textStyle="bold"
2423
app:layout_constraintBottom_toTopOf="@id/clear_button"
2524
app:layout_constraintLeft_toLeftOf="parent"
2625
app:layout_constraintRight_toRightOf="parent"
27-
app:layout_constraintTop_toBottomOf="@id/draw_view"/>
26+
app:layout_constraintTop_toBottomOf="@id/draw_view" />
2827

2928
<Button
3029
android:id="@+id/clear_button"
@@ -33,6 +32,15 @@
3332
android:text="@string/clear_button_text"
3433
app:layout_constraintBottom_toBottomOf="parent"
3534
app:layout_constraintLeft_toLeftOf="parent"
36-
app:layout_constraintRight_toRightOf="parent"/>
35+
app:layout_constraintRight_toRightOf="parent" />
36+
37+
<Button
38+
android:id="@+id/init_model"
39+
android:layout_width="wrap_content"
40+
android:layout_height="wrap_content"
41+
android:text="init_model"
42+
app:layout_constraintBottom_toBottomOf="parent"
43+
app:layout_constraintLeft_toLeftOf="parent"
44+
app:layout_constraintStart_toEndOf="@id/clear_button" />
3745

3846
</androidx.constraintlayout.widget.ConstraintLayout>

0 commit comments

Comments
 (0)