Skip to content

R
RND_Camera_Vision
Commit 2ccd44f5 authored by Syazmin's avatar Syazmin
Browse files
Options

Initial people detection camera app

parents
Show whitespace changes
Inline Side-by-side
Showing with 2278 additions and 0 deletions
.gitignore 0 → 100644
View file @ 2ccd44f5
.gradle/
.idea/
**/build/
local.properties
*.iml
.externalNativeBuild/
.cxx/
captures/
app/src/main/assets/*.tflite
app/src/androidTest/assets/*.tflite
README.md 0 → 100644
View file @ 2ccd44f5
# TensorFlow Lite Object Detection Android Demo
### Overview
This is a camera app that continuously detects the objects (bounding boxes and
classes) in the frames seen by your device's back camera, with the option to use
a quantized
[MobileNet SSD](https://tfhub.dev/tensorflow/lite-model/ssd_mobilenet_v1/1/metadata/2),
[EfficientDet Lite 0](https://tfhub.dev/tensorflow/lite-model/efficientdet/lite0/detection/metadata/1),
[EfficientDet Lite1](https://tfhub.dev/tensorflow/lite-model/efficientdet/lite1/detection/metadata/1),
or
[EfficientDet Lite2](https://tfhub.dev/tensorflow/lite-model/efficientdet/lite2/detection/metadata/1)
model trained on the [COCO dataset](http://cocodataset.org/). These instructions
walk you through building and running the demo on an Android device.
The model files are downloaded via Gradle scripts when you build and run the
app. You don't need to do any steps to download TFLite models into the project
explicitly.
This application should be run on a physical Android device.
![App example showing UI controls. Highlights a cat](https://storage.googleapis.com/download.tensorflow.org/tflite/examples/obj_detection_cat.gif)
![App example showing UI controls. Highlights a cat, a book, and a couch.](screenshot1.png)
## Build the demo using Android Studio
### Prerequisites
* The **[Android Studio](https://developer.android.com/studio/index.html)**
IDE. This sample has been tested on Android Studio Bumblebee.
* A physical Android device with a minimum OS version of SDK 24 (Android 7.0 -
Nougat) with developer mode enabled. The process of enabling developer mode
may vary by device.
### Building
* Open Android Studio. From the Welcome screen, select Open an existing
Android Studio project.
* From the Open File or Project window that appears, navigate to and select
the tensorflow-lite/examples/object_detection/android directory. Click OK.
* If it asks you to do a Gradle Sync, click OK.
* With your Android device connected to your computer and developer mode
enabled, click on the green Run arrow in Android Studio.
### Models used
Downloading, extraction, and placing the models into the assets folder is
managed automatically by the download.gradle file.
app/build.gradle 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* https://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
plugins {
id 'com.android.application'
id 'kotlin-android'
id 'kotlin-kapt'
id 'androidx.navigation.safeargs'
id 'de.undercouch.download'
}
android {
namespace "org.tensorflow.lite.examples.objectdetection"
compileSdk 34
defaultConfig {
applicationId "org.tensorflow.lite.examples.objectdetection"
minSdk 21
targetSdk 34
versionCode 1
versionName "1.0.0"
testInstrumentationRunner "androidx.test.runner.AndroidJUnitRunner"
}
buildFeatures {
dataBinding true
viewBinding true
}
compileOptions {
sourceCompatibility JavaVersion.VERSION_17
targetCompatibility JavaVersion.VERSION_17
}
kotlinOptions {
jvmTarget = "17"
}
buildTypes {
release {
minifyEnabled false
proguardFiles getDefaultProguardFile('proguard-android.txt'), 'proguard-rules.pro'
}
}
androidResources {
noCompress 'tflite'
}
}
// import DownloadModels task
project.ext.ASSET_DIR = projectDir.toString() + '/src/main/assets'
project.ext.TEST_ASSETS_DIR = projectDir.toString() + '/src/androidTest/assets'
// Download default models; if you wish to use your own models then
// place them in the "assets" directory and comment out this line.
apply from:'download_models.gradle'
dependencies {
// Kotlin lang
implementation 'androidx.core:core-ktx:1.9.0'
implementation "org.jetbrains.kotlin:kotlin-stdlib:1.9.22"
implementation 'org.jetbrains.kotlinx:kotlinx-coroutines-android:1.7.3'
// App compat and UI things
implementation 'androidx.appcompat:appcompat:1.6.1'
implementation 'androidx.lifecycle:lifecycle-runtime-ktx:2.6.2'
implementation 'androidx.constraintlayout:constraintlayout:2.1.4'
implementation 'com.google.android.material:material:1.11.0'
implementation 'androidx.localbroadcastmanager:localbroadcastmanager:1.1.0'
// Navigation library
def nav_version = "2.7.7"
implementation "androidx.navigation:navigation-fragment-ktx:$nav_version"
implementation "androidx.navigation:navigation-ui-ktx:$nav_version"
// CameraX core library
def camerax_version = '1.3.1'
implementation "androidx.camera:camera-core:$camerax_version"
implementation "androidx.camera:camera-camera2:$camerax_version"
implementation "androidx.camera:camera-lifecycle:$camerax_version"
implementation "androidx.camera:camera-view:$camerax_version"
//WindowManager
implementation 'androidx.window:window:1.2.0'
// Unit testing
testImplementation 'androidx.test.ext:junit:1.1.5'
testImplementation 'androidx.test:rules:1.5.0'
testImplementation 'androidx.test:runner:1.5.2'
testImplementation 'androidx.test.espresso:espresso-core:3.5.1'
testImplementation 'org.robolectric:robolectric:4.11.1'
// Instrumented testing
androidTestImplementation "androidx.test.ext:junit:1.1.5"
androidTestImplementation "androidx.test:core:1.5.0"
androidTestImplementation "androidx.test:rules:1.5.0"
androidTestImplementation "androidx.test:runner:1.5.2"
androidTestImplementation "androidx.test.espresso:espresso-core:3.5.1"
implementation 'org.tensorflow:tensorflow-lite-task-vision:0.4.4'
// Import the GPU delegate plugin Library for GPU inference
implementation 'org.tensorflow:tensorflow-lite-gpu-delegate-plugin:0.4.4'
implementation 'org.tensorflow:tensorflow-lite-gpu:2.14.0'
}
app/download_models.gradle 0 → 100644
View file @ 2ccd44f5
task downloadModelFile(type: Download) {
src 'https://storage.googleapis.com/download.tensorflow.org/models/tflite/task_library/object_detection/android/lite-model_ssd_mobilenet_v1_1_metadata_2.tflite'
dest project.ext.ASSET_DIR + '/mobilenetv1.tflite'
overwrite false
}
task downloadModelFile0(type: Download) {
src 'https://storage.googleapis.com/download.tensorflow.org/models/tflite/task_library/object_detection/android/lite-model_efficientdet_lite0_detection_metadata_1.tflite'
dest project.ext.ASSET_DIR + '/efficientdet-lite0.tflite'
overwrite false
}
task downloadModelFile1(type: Download) {
src 'https://storage.googleapis.com/download.tensorflow.org/models/tflite/task_library/object_detection/android/lite-model_efficientdet_lite1_detection_metadata_1.tflite'
dest project.ext.ASSET_DIR + '/efficientdet-lite1.tflite'
overwrite false
}
task downloadModelFile2(type: Download) {
src 'https://storage.googleapis.com/download.tensorflow.org/models/tflite/task_library/object_detection/android/lite-model_efficientdet_lite2_detection_metadata_1.tflite'
dest project.ext.ASSET_DIR + '/efficientdet-lite2.tflite'
overwrite false
}
task copyTestModel(type: Copy, dependsOn: downloadModelFile) {
from project.ext.ASSET_DIR + '/mobilenetv1.tflite'
into project.ext.TEST_ASSETS_DIR
}
preBuild.dependsOn downloadModelFile, downloadModelFile0, downloadModelFile1, downloadModelFile2,
copyTestModel
\ No newline at end of file
app/proguard-rules.pro 0 → 100644
View file @ 2ccd44f5
# Add project specific ProGuard rules here.
# You can control the set of applied configuration files using the
# proguardFiles setting in build.gradle.
#
# For more details, see
# http://developer.android.com/guide/developing/tools/proguard.html
# If your project uses WebView with JS, uncomment the following
# and specify the fully qualified class name to the JavaScript interface
# class:
#-keepclassmembers class fqcn.of.javascript.interface.for.webview {
# public *;
#}
# Uncomment this to preserve the line number information for
# debugging stack traces.
#-keepattributes SourceFile,LineNumberTable
# If you keep the line number information, uncomment this to
# hide the original source file name.
#-renamesourcefileattribute SourceFile
\ No newline at end of file
app/src/androidTest/java/org/tensorflow/lite/examples/objectdetection/TFObjectDetectionTest.kt 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.tensorflow.lite.examples.objectdetection
import android.content.res.AssetManager
import android.graphics.Bitmap
import android.graphics.BitmapFactory
import android.graphics.RectF
import androidx.test.ext.junit.runners.AndroidJUnit4
import androidx.test.platform.app.InstrumentationRegistry
import java.io.InputStream
import org.junit.Assert.assertEquals
import org.junit.Assert.assertNotNull
import org.junit.Assert.assertTrue
import org.junit.Test
import org.junit.runner.RunWith
import org.tensorflow.lite.support.label.Category
import org.tensorflow.lite.task.vision.detector.Detection
/**
* Instrumented test, which will execute on an Android device.
*
* See [testing documentation](http://d.android.com/tools/testing).
*/
@RunWith(AndroidJUnit4::class)
class TFObjectDetectionTest {
val controlResults = listOf<Detection>(
Detection.create(RectF(69.0f, 58.0f, 227.0f, 171.0f),
listOf<Category>(Category.create("cat", "cat", 0.77734375f))),
Detection.create(RectF(13.0f, 6.0f, 283.0f, 215.0f),
listOf<Category>(Category.create("couch", "couch", 0.5859375f))),
Detection.create(RectF(45.0f, 27.0f, 257.0f, 184.0f),
listOf<Category>(Category.create("chair", "chair", 0.55078125f)))
)
@Test
@Throws(Exception::class)
fun detectionResultsShouldNotChange() {
val objectDetectorHelper =
ObjectDetectorHelper(
context = InstrumentationRegistry.getInstrumentation().context,
objectDetectorListener =
object : ObjectDetectorHelper.DetectorListener {
override fun onError(error: String) {
// no op
}
override fun onResults(
results: MutableList<Detection>?,
inferenceTime: Long,
imageHeight: Int,
imageWidth: Int
) {
assertEquals(controlResults.size, results!!.size)
// Loop through the detected and control data
for (i in controlResults.indices) {
// Verify that the bounding boxes are the same
assertEquals(results[i].boundingBox, controlResults[i].boundingBox)
// Verify that the detected data and control
// data have the same number of categories
assertEquals(
results[i].categories.size,
controlResults[i].categories.size
)
// Loop through the categories
for (j in 0 until controlResults[i].categories.size - 1) {
// Verify that the labels are consistent
assertEquals(
results[i].categories[j].label,
controlResults[i].categories[j].label
)
}
}
}
}
)
// Create Bitmap and convert to TensorImage
val bitmap = loadImage("cat1.png")
// Run the object detector on the sample image
objectDetectorHelper.detect(bitmap!!, 0)
}
@Test
@Throws(Exception::class)
fun detectedImageIsScaledWithinModelDimens() {
val objectDetectorHelper =
ObjectDetectorHelper(
context = InstrumentationRegistry.getInstrumentation().context,
objectDetectorListener =
object : ObjectDetectorHelper.DetectorListener {
override fun onError(error: String) {}
override fun onResults(
results: MutableList<Detection>?,
inferenceTime: Long,
imageHeight: Int,
imageWidth: Int
) {
assertNotNull(results)
for (result in results!!) {
assertTrue(result.boundingBox.top <= imageHeight)
assertTrue(result.boundingBox.bottom <= imageHeight)
assertTrue(result.boundingBox.left <= imageWidth)
assertTrue(result.boundingBox.right <= imageWidth)
}
}
}
)
// Create Bitmap and convert to TensorImage
val bitmap = loadImage("cat1.png")
// Run the object detector on the sample image
objectDetectorHelper.detect(bitmap!!, 0)
}
@Throws(Exception::class)
private fun loadImage(fileName: String): Bitmap? {
val assetManager: AssetManager =
InstrumentationRegistry.getInstrumentation().context.assets
val inputStream: InputStream = assetManager.open(fileName)
return BitmapFactory.decodeStream(inputStream)
}
}
app/src/main/AndroidManifest.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<manifest
xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:dist="http://schemas.android.com/apk/distribution"
xmlns:tools="http://schemas.android.com/tools"
package="org.tensorflow.lite.examples.objectdetection">
<!-- Enable instant app support -->
<dist:module dist:instant="true" />
<!-- Declare features -->
<uses-feature android:name="android.hardware.camera" />
<!-- Declare permissions -->
<uses-permission android:name="android.permission.CAMERA" />
<application
android:icon="@mipmap/ic_launcher"
android:roundIcon="@mipmap/ic_launcher_round"
android:label="@string/app_name"
android:allowBackup="true"
android:taskAffinity=""
tools:ignore="AllowBackup">
<activity
android:name=".MainActivity"
android:clearTaskOnLaunch="true"
android:theme="@style/AppTheme"
android:exported="true"
android:icon="@mipmap/ic_launcher"
android:rotationAnimation="seamless"
android:resizeableActivity="true"
android:configChanges="orientation|screenLayout|screenSize|smallestScreenSize"
tools:targetApi="O">
<!-- Main app intent filter -->
<intent-filter>
<action android:name="android.intent.action.MAIN" />
<category android:name="android.intent.category.LAUNCHER" />
</intent-filter>
<!-- Declare notch support -->
<meta-data android:name="android.notch_support" android:value="true"/>
</activity>
</application>
</manifest>
app/src/main/java/org/tensorflow/lite/examples/objectdetection/MainActivity.kt 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.tensorflow.lite.examples.objectdetection
import android.os.Build
import android.os.Bundle
import androidx.appcompat.app.AppCompatActivity
import org.tensorflow.lite.examples.objectdetection.databinding.ActivityMainBinding
/**
* Main entry point into our app. This app follows the single-activity pattern, and all
* functionality is implemented in the form of fragments.
*/
class MainActivity : AppCompatActivity() {
private lateinit var activityMainBinding: ActivityMainBinding
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
activityMainBinding = ActivityMainBinding.inflate(layoutInflater)
setContentView(activityMainBinding.root)
}
override fun onBackPressed() {
if (Build.VERSION.SDK_INT == Build.VERSION_CODES.Q) {
// Workaround for Android Q memory leak issue in IRequestFinishCallback$Stub.
// (https://issuetracker.google.com/issues/139738913)
finishAfterTransition()
} else {
super.onBackPressed()
}
}
}
app/src/main/java/org/tensorflow/lite/examples/objectdetection/ObjectDetectorHelper.kt 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.tensorflow.lite.examples.objectdetection
import android.content.Context
import android.graphics.Bitmap
import android.os.SystemClock
import android.util.Log
import org.tensorflow.lite.gpu.CompatibilityList
import org.tensorflow.lite.support.image.ImageProcessor
import org.tensorflow.lite.support.image.TensorImage
import org.tensorflow.lite.support.image.ops.Rot90Op
import org.tensorflow.lite.task.core.BaseOptions
import org.tensorflow.lite.task.vision.detector.Detection
import org.tensorflow.lite.task.vision.detector.ObjectDetector
class ObjectDetectorHelper(
var threshold: Float = 0.5f,
var numThreads: Int = 2,
var maxResults: Int = 20,
var currentDelegate: Int = 0,
var currentModel: Int = 0,
val context: Context,
val objectDetectorListener: DetectorListener?
) {
// For this example this needs to be a var so it can be reset on changes. If the ObjectDetector
// will not change, a lazy val would be preferable.
private var objectDetector: ObjectDetector? = null
init {
setupObjectDetector()
}
fun clearObjectDetector() {
objectDetector = null
}
// Initialize the object detector using current settings on the
// thread that is using it. CPU and NNAPI delegates can be used with detectors
// that are created on the main thread and used on a background thread, but
// the GPU delegate needs to be used on the thread that initialized the detector
fun setupObjectDetector() {
// Create the base options for the detector using specifies max results and score threshold
val optionsBuilder =
ObjectDetector.ObjectDetectorOptions.builder()
.setScoreThreshold(threshold)
// Set general detection options, including number of used threads
val baseOptionsBuilder = BaseOptions.builder().setNumThreads(numThreads)
// Use the specified hardware for running the model. Default to CPU
when (currentDelegate) {
DELEGATE_CPU -> {
// Default
}
DELEGATE_GPU -> {
if (CompatibilityList().isDelegateSupportedOnThisDevice) {
baseOptionsBuilder.useGpu()
} else {
objectDetectorListener?.onError("GPU is not supported on this device")
}
}
DELEGATE_NNAPI -> {
baseOptionsBuilder.useNnapi()
}
}
optionsBuilder.setBaseOptions(baseOptionsBuilder.build())
val modelName =
when (currentModel) {
MODEL_MOBILENETV1 -> "mobilenetv1.tflite"
MODEL_EFFICIENTDETV0 -> "efficientdet-lite0.tflite"
MODEL_EFFICIENTDETV1 -> "efficientdet-lite1.tflite"
MODEL_EFFICIENTDETV2 -> "efficientdet-lite2.tflite"
else -> "mobilenetv1.tflite"
}
try {
objectDetector =
ObjectDetector.createFromFileAndOptions(context, modelName, optionsBuilder.build())
} catch (e: IllegalStateException) {
objectDetectorListener?.onError(
"Object detector failed to initialize. See error logs for details"
)
Log.e("Test", "TFLite failed to load model with error: " + e.message)
}
}
fun detect(image: Bitmap, imageRotation: Int) {
if (objectDetector == null) {
setupObjectDetector()
}
// Inference time is the difference between the system time at the start and finish of the
// process
var inferenceTime = SystemClock.uptimeMillis()
// Create preprocessor for the image.
// See https://www.tensorflow.org/lite/inference_with_metadata/
// lite_support#imageprocessor_architecture
val imageProcessor =
ImageProcessor.Builder()
.add(Rot90Op(-imageRotation / 90))
.build()
// Preprocess the image and convert it into a TensorImage for detection.
val tensorImage = imageProcessor.process(TensorImage.fromBitmap(image))
// This Task Vision version has no category allowlist. Request all model results,
// then keep only people before applying the user-selected display limit.
val results = objectDetector
?.detect(tensorImage)
?.filter { detection ->
detection.categories.any {
it.label.equals(TARGET_CATEGORY, ignoreCase = true)
}
}
?.take(maxResults)
?.toMutableList()
inferenceTime = SystemClock.uptimeMillis() - inferenceTime
objectDetectorListener?.onResults(
results,
inferenceTime,
tensorImage.height,
tensorImage.width)
}
interface DetectorListener {
fun onError(error: String)
fun onResults(
results: MutableList<Detection>?,
inferenceTime: Long,
imageHeight: Int,
imageWidth: Int
)
}
companion object {
const val DELEGATE_CPU = 0
const val DELEGATE_GPU = 1
const val DELEGATE_NNAPI = 2
const val MODEL_MOBILENETV1 = 0
const val MODEL_EFFICIENTDETV0 = 1
const val MODEL_EFFICIENTDETV1 = 2
const val MODEL_EFFICIENTDETV2 = 3
const val MAX_RESULTS_LIMIT = 20
const val TARGET_CATEGORY = "person"
}
}
app/src/main/java/org/tensorflow/lite/examples/objectdetection/OverlayView.kt 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.tensorflow.lite.examples.objectdetection
import android.content.Context
import android.graphics.Canvas
import android.graphics.Color
import android.graphics.Paint
import android.graphics.Rect
import android.graphics.RectF
import android.util.AttributeSet
import android.view.View
import androidx.core.content.ContextCompat
import java.util.LinkedList
import kotlin.math.max
import org.tensorflow.lite.task.vision.detector.Detection
class OverlayView(context: Context?, attrs: AttributeSet?) : View(context, attrs) {
private var results: List<Detection> = LinkedList<Detection>()
private var boxPaint = Paint()
private var textBackgroundPaint = Paint()
private var textPaint = Paint()
private var scaleFactor: Float = 1f
private var bounds = Rect()
init {
initPaints()
}
fun clear() {
textPaint.reset()
textBackgroundPaint.reset()
boxPaint.reset()
invalidate()
initPaints()
}
private fun initPaints() {
textBackgroundPaint.color = Color.BLACK
textBackgroundPaint.style = Paint.Style.FILL
textBackgroundPaint.textSize = 50f
textPaint.color = Color.WHITE
textPaint.style = Paint.Style.FILL
textPaint.textSize = 50f
boxPaint.color = ContextCompat.getColor(context!!, R.color.bounding_box_color)
boxPaint.strokeWidth = 8F
boxPaint.style = Paint.Style.STROKE
}
override fun draw(canvas: Canvas) {
super.draw(canvas)
for (result in results) {
val boundingBox = result.boundingBox
val top = boundingBox.top * scaleFactor
val bottom = boundingBox.bottom * scaleFactor
val left = boundingBox.left * scaleFactor
val right = boundingBox.right * scaleFactor
// Draw bounding box around detected objects
val drawableRect = RectF(left, top, right, bottom)
canvas.drawRect(drawableRect, boxPaint)
// Create text to display alongside detected objects
val drawableText =
result.categories[0].label + " " +
String.format("%.2f", result.categories[0].score)
// Draw rect behind display text
textBackgroundPaint.getTextBounds(drawableText, 0, drawableText.length, bounds)
val textWidth = bounds.width()
val textHeight = bounds.height()
canvas.drawRect(
left,
top,
left + textWidth + Companion.BOUNDING_RECT_TEXT_PADDING,
top + textHeight + Companion.BOUNDING_RECT_TEXT_PADDING,
textBackgroundPaint
)
// Draw text for detected object
canvas.drawText(drawableText, left, top + bounds.height(), textPaint)
}
}
fun setResults(
detectionResults: MutableList<Detection>,
imageHeight: Int,
imageWidth: Int,
) {
results = detectionResults
// PreviewView is in FILL_START mode. So we need to scale up the bounding box to match with
// the size that the captured images will be displayed.
scaleFactor = max(width * 1f / imageWidth, height * 1f / imageHeight)
}
companion object {
private const val BOUNDING_RECT_TEXT_PADDING = 8
}
}
app/src/main/java/org/tensorflow/lite/examples/objectdetection/fragments/CameraFragment.kt 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.tensorflow.lite.examples.objectdetection.fragments
import android.annotation.SuppressLint
import android.content.res.Configuration
import android.graphics.Bitmap
import android.os.Bundle
import android.util.Log
import android.view.LayoutInflater
import android.view.View
import android.view.ViewGroup
import android.widget.AdapterView
import android.widget.Toast
import androidx.camera.core.AspectRatio
import androidx.camera.core.Camera
import androidx.camera.core.CameraSelector
import androidx.camera.core.ImageAnalysis
import androidx.camera.core.ImageAnalysis.OUTPUT_IMAGE_FORMAT_RGBA_8888
import androidx.camera.core.ImageProxy
import androidx.camera.core.Preview
import androidx.camera.lifecycle.ProcessCameraProvider
import androidx.core.content.ContextCompat
import androidx.fragment.app.Fragment
import androidx.navigation.Navigation
import java.util.LinkedList
import java.util.concurrent.ExecutorService
import java.util.concurrent.Executors
import org.tensorflow.lite.examples.objectdetection.ObjectDetectorHelper
import org.tensorflow.lite.examples.objectdetection.R
import org.tensorflow.lite.examples.objectdetection.databinding.FragmentCameraBinding
import org.tensorflow.lite.task.vision.detector.Detection
class CameraFragment : Fragment(), ObjectDetectorHelper.DetectorListener {
private val TAG = "ObjectDetection"
private var _fragmentCameraBinding: FragmentCameraBinding? = null
private val fragmentCameraBinding
get() = _fragmentCameraBinding!!
private lateinit var objectDetectorHelper: ObjectDetectorHelper
private lateinit var bitmapBuffer: Bitmap
private var preview: Preview? = null
private var imageAnalyzer: ImageAnalysis? = null
private var camera: Camera? = null
private var cameraProvider: ProcessCameraProvider? = null
/** Blocking camera operations are performed using this executor */
private lateinit var cameraExecutor: ExecutorService
override fun onResume() {
super.onResume()
// Make sure that all permissions are still present, since the
// user could have removed them while the app was in paused state.
if (!PermissionsFragment.hasPermissions(requireContext())) {
Navigation.findNavController(requireActivity(), R.id.fragment_container)
.navigate(CameraFragmentDirections.actionCameraToPermissions())
}
}
override fun onDestroyView() {
_fragmentCameraBinding = null
super.onDestroyView()
// Shut down our background executor
cameraExecutor.shutdown()
}
override fun onCreateView(
inflater: LayoutInflater,
container: ViewGroup?,
savedInstanceState: Bundle?
): View {
_fragmentCameraBinding = FragmentCameraBinding.inflate(inflater, container, false)
return fragmentCameraBinding.root
}
@SuppressLint("MissingPermission")
override fun onViewCreated(view: View, savedInstanceState: Bundle?) {
super.onViewCreated(view, savedInstanceState)
objectDetectorHelper = ObjectDetectorHelper(
context = requireContext(),
objectDetectorListener = this)
// Initialize our background executor
cameraExecutor = Executors.newSingleThreadExecutor()
// Wait for the views to be properly laid out
fragmentCameraBinding.viewFinder.post {
// Set up the camera and its use cases
setUpCamera()
}
// Attach listeners to UI control widgets
initBottomSheetControls()
}
private fun initBottomSheetControls() {
// When clicked, lower detection score threshold floor
fragmentCameraBinding.bottomSheetLayout.thresholdMinus.setOnClickListener {
if (objectDetectorHelper.threshold >= 0.1) {
objectDetectorHelper.threshold -= 0.1f
updateControlsUi()
}
}
// When clicked, raise detection score threshold floor
fragmentCameraBinding.bottomSheetLayout.thresholdPlus.setOnClickListener {
if (objectDetectorHelper.threshold <= 0.8) {
objectDetectorHelper.threshold += 0.1f
updateControlsUi()
}
}
// When clicked, reduce the number of objects that can be detected at a time
fragmentCameraBinding.bottomSheetLayout.maxResultsMinus.setOnClickListener {
if (objectDetectorHelper.maxResults > 1) {
objectDetectorHelper.maxResults--
updateControlsUi()
}
}
// When clicked, increase the number of objects that can be detected at a time
fragmentCameraBinding.bottomSheetLayout.maxResultsPlus.setOnClickListener {
if (objectDetectorHelper.maxResults < ObjectDetectorHelper.MAX_RESULTS_LIMIT) {
objectDetectorHelper.maxResults++
updateControlsUi()
}
}
// When clicked, decrease the number of threads used for detection
fragmentCameraBinding.bottomSheetLayout.threadsMinus.setOnClickListener {
if (objectDetectorHelper.numThreads > 1) {
objectDetectorHelper.numThreads--
updateControlsUi()
}
}
// When clicked, increase the number of threads used for detection
fragmentCameraBinding.bottomSheetLayout.threadsPlus.setOnClickListener {
if (objectDetectorHelper.numThreads < 4) {
objectDetectorHelper.numThreads++
updateControlsUi()
}
}
// When clicked, change the underlying hardware used for inference. Current options are CPU
// GPU, and NNAPI
fragmentCameraBinding.bottomSheetLayout.spinnerDelegate.setSelection(0, false)
fragmentCameraBinding.bottomSheetLayout.spinnerDelegate.onItemSelectedListener =
object : AdapterView.OnItemSelectedListener {
override fun onItemSelected(p0: AdapterView<*>?, p1: View?, p2: Int, p3: Long) {
objectDetectorHelper.currentDelegate = p2
updateControlsUi()
}
override fun onNothingSelected(p0: AdapterView<*>?) {
/* no op */
}
}
// When clicked, change the underlying model used for object detection
fragmentCameraBinding.bottomSheetLayout.spinnerModel.setSelection(0, false)
fragmentCameraBinding.bottomSheetLayout.spinnerModel.onItemSelectedListener =
object : AdapterView.OnItemSelectedListener {
override fun onItemSelected(p0: AdapterView<*>?, p1: View?, p2: Int, p3: Long) {
objectDetectorHelper.currentModel = p2
updateControlsUi()
}
override fun onNothingSelected(p0: AdapterView<*>?) {
/* no op */
}
}
}
// Update the values displayed in the bottom sheet. Reset detector.
private fun updateControlsUi() {
fragmentCameraBinding.bottomSheetLayout.maxResultsValue.text =
objectDetectorHelper.maxResults.toString()
fragmentCameraBinding.bottomSheetLayout.thresholdValue.text =
String.format("%.2f", objectDetectorHelper.threshold)
fragmentCameraBinding.bottomSheetLayout.threadsValue.text =
objectDetectorHelper.numThreads.toString()
// Needs to be cleared instead of reinitialized because the GPU
// delegate needs to be initialized on the thread using it when applicable
objectDetectorHelper.clearObjectDetector()
fragmentCameraBinding.overlay.clear()
fragmentCameraBinding.bottomSheetLayout.detectedCountValue.text = "0"
}
// Initialize CameraX, and prepare to bind the camera use cases
private fun setUpCamera() {
val cameraProviderFuture = ProcessCameraProvider.getInstance(requireContext())
cameraProviderFuture.addListener(
{
// CameraProvider
cameraProvider = cameraProviderFuture.get()
// Build and bind the camera use cases
bindCameraUseCases()
},
ContextCompat.getMainExecutor(requireContext())
)
}
// Declare and bind preview, capture and analysis use cases
@SuppressLint("UnsafeOptInUsageError")
private fun bindCameraUseCases() {
// CameraProvider
val cameraProvider =
cameraProvider ?: throw IllegalStateException("Camera initialization failed.")
// CameraSelector - makes assumption that we're only using the back camera
val cameraSelector =
CameraSelector.Builder().requireLensFacing(CameraSelector.LENS_FACING_BACK).build()
// Preview. Only using the 4:3 ratio because this is the closest to our models
preview =
Preview.Builder()
.setTargetAspectRatio(AspectRatio.RATIO_4_3)
.setTargetRotation(fragmentCameraBinding.viewFinder.display.rotation)
.build()
// ImageAnalysis. Using RGBA 8888 to match how our models work
imageAnalyzer =
ImageAnalysis.Builder()
.setTargetAspectRatio(AspectRatio.RATIO_4_3)
.setTargetRotation(fragmentCameraBinding.viewFinder.display.rotation)
.setBackpressureStrategy(ImageAnalysis.STRATEGY_KEEP_ONLY_LATEST)
.setOutputImageFormat(OUTPUT_IMAGE_FORMAT_RGBA_8888)
.build()
// The analyzer can then be assigned to the instance
.also {
it.setAnalyzer(cameraExecutor) { image ->
if (!::bitmapBuffer.isInitialized) {
// The image rotation and RGB image buffer are initialized only once
// the analyzer has started running
bitmapBuffer = Bitmap.createBitmap(
image.width,
image.height,
Bitmap.Config.ARGB_8888
)
}
detectObjects(image)
}
}
// Must unbind the use-cases before rebinding them
cameraProvider.unbindAll()
try {
// A variable number of use-cases can be passed here -
// camera provides access to CameraControl & CameraInfo
camera = cameraProvider.bindToLifecycle(this, cameraSelector, preview, imageAnalyzer)
// Attach the viewfinder's surface provider to preview use case
preview?.setSurfaceProvider(fragmentCameraBinding.viewFinder.surfaceProvider)
} catch (exc: Exception) {
Log.e(TAG, "Use case binding failed", exc)
}
}
private fun detectObjects(image: ImageProxy) {
// Copy out RGB bits to the shared bitmap buffer
image.use { bitmapBuffer.copyPixelsFromBuffer(image.planes[0].buffer) }
val imageRotation = image.imageInfo.rotationDegrees
// Pass Bitmap and rotation to the object detector helper for processing and detection
objectDetectorHelper.detect(bitmapBuffer, imageRotation)
}
override fun onConfigurationChanged(newConfig: Configuration) {
super.onConfigurationChanged(newConfig)
imageAnalyzer?.targetRotation = fragmentCameraBinding.viewFinder.display.rotation
}
// Update UI after objects have been detected. Extracts original image height/width
// to scale and place bounding boxes properly through OverlayView
override fun onResults(
results: MutableList<Detection>?,
inferenceTime: Long,
imageHeight: Int,
imageWidth: Int
) {
activity?.runOnUiThread {
val people = results
?.filter { detection ->
detection.categories.any {
it.label.equals(ObjectDetectorHelper.TARGET_CATEGORY, ignoreCase = true)
}
}
?.toMutableList()
?: LinkedList()
fragmentCameraBinding.bottomSheetLayout.inferenceTimeVal.text =
String.format("%d ms", inferenceTime)
fragmentCameraBinding.bottomSheetLayout.detectedCountValue.text =
people.size.toString()
// Pass necessary information to OverlayView for drawing on the canvas
fragmentCameraBinding.overlay.setResults(
people,
imageHeight,
imageWidth
)
// Force a redraw
fragmentCameraBinding.overlay.invalidate()
}
}
override fun onError(error: String) {
activity?.runOnUiThread {
Toast.makeText(requireContext(), error, Toast.LENGTH_SHORT).show()
}
}
}
app/src/main/java/org/tensorflow/lite/examples/objectdetection/fragments/PermissionsFragment.kt 0 → 100644
View file @ 2ccd44f5
/*
* Copyright 2022 The TensorFlow Authors. All Rights Reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package org.tensorflow.lite.examples.objectdetection.fragments
import android.Manifest
import android.content.Context
import android.content.pm.PackageManager
import android.os.Bundle
import android.widget.Toast
import androidx.activity.result.contract.ActivityResultContracts
import androidx.core.content.ContextCompat
import androidx.fragment.app.Fragment
import androidx.lifecycle.lifecycleScope
import androidx.navigation.Navigation
import org.tensorflow.lite.examples.objectdetection.R
private val PERMISSIONS_REQUIRED = arrayOf(Manifest.permission.CAMERA)
/**
* The sole purpose of this fragment is to request permissions and, once granted, display the
* camera fragment to the user.
*/
class PermissionsFragment : Fragment() {
private val requestPermissionLauncher =
registerForActivityResult(ActivityResultContracts.RequestPermission()
) { isGranted: Boolean ->
if (isGranted) {
Toast.makeText(context, "Permission request granted", Toast.LENGTH_LONG).show()
navigateToCamera()
} else {
Toast.makeText(context, "Permission request denied", Toast.LENGTH_LONG).show()
}
}
override fun onCreate(savedInstanceState: Bundle?) {
super.onCreate(savedInstanceState)
when {
ContextCompat.checkSelfPermission(
requireContext(),
Manifest.permission.CAMERA
) == PackageManager.PERMISSION_GRANTED -> {
navigateToCamera()
}
else -> {
requestPermissionLauncher.launch(
Manifest.permission.CAMERA)
}
}
}
private fun navigateToCamera() {
lifecycleScope.launchWhenStarted {
Navigation.findNavController(requireActivity(), R.id.fragment_container).navigate(
PermissionsFragmentDirections.actionPermissionsToCamera())
}
}
companion object {
/** Convenience method used to check if all permissions required by this app are granted */
fun hasPermissions(context: Context) = PERMISSIONS_REQUIRED.all {
ContextCompat.checkSelfPermission(context, it) == PackageManager.PERMISSION_GRANTED
}
}
}
app/src/main/res/color/selector_ic.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<selector xmlns:android="http://schemas.android.com/apk/res/android">
<item android:state_pressed="true" android:color="@color/icPressed" />
<item android:state_focused="true" android:color="@color/icFocused" />
<item android:color="@color/icActive" />
</selector>
\ No newline at end of file
app/src/main/res/drawable-v24/ic_launcher_foreground.xml 0 → 100644
View file @ 2ccd44f5
<vector xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:aapt="http://schemas.android.com/aapt"
android:width="108dp"
android:height="108dp"
android:viewportWidth="108"
android:viewportHeight="108">
<path android:pathData="M31,63.928c0,0 6.4,-11 12.1,-13.1c7.2,-2.6 26,-1.4 26,-1.4l38.1,38.1L107,108.928l-32,-1L31,63.928z">
<aapt:attr name="android:fillColor">
<gradient
android:endX="85.84757"
android:endY="92.4963"
android:startX="42.9492"
android:startY="49.59793"
android:type="linear">
<item
android:color="#44000000"
android:offset="0.0" />
<item
android:color="#00000000"
android:offset="1.0" />
</gradient>
</aapt:attr>
</path>
<path
android:fillColor="#FFFFFF"
android:fillType="nonZero"
android:pathData="M65.3,45.828l3.8,-6.6c0.2,-0.4 0.1,-0.9 -0.3,-1.1c-0.4,-0.2 -0.9,-0.1 -1.1,0.3l-3.9,6.7c-6.3,-2.8 -13.4,-2.8 -19.7,0l-3.9,-6.7c-0.2,-0.4 -0.7,-0.5 -1.1,-0.3C38.8,38.328 38.7,38.828 38.9,39.228l3.8,6.6C36.2,49.428 31.7,56.028 31,63.928h46C76.3,56.028 71.8,49.428 65.3,45.828zM43.4,57.328c-0.8,0 -1.5,-0.5 -1.8,-1.2c-0.3,-0.7 -0.1,-1.5 0.4,-2.1c0.5,-0.5 1.4,-0.7 2.1,-0.4c0.7,0.3 1.2,1 1.2,1.8C45.3,56.528 44.5,57.328 43.4,57.328L43.4,57.328zM64.6,57.328c-0.8,0 -1.5,-0.5 -1.8,-1.2s-0.1,-1.5 0.4,-2.1c0.5,-0.5 1.4,-0.7 2.1,-0.4c0.7,0.3 1.2,1 1.2,1.8C66.5,56.528 65.6,57.328 64.6,57.328L64.6,57.328z"
android:strokeWidth="1"
android:strokeColor="#00000000" />
</vector>
\ No newline at end of file
app/src/main/res/drawable/ic_minus.xml 0 → 100644
View file @ 2ccd44f5
<vector xmlns:android="http://schemas.android.com/apk/res/android"
android:width="24dp"
android:height="24dp"
android:viewportWidth="24"
android:viewportHeight="24">
<path
android:fillColor="#FF000000"
android:pathData="M19,13H5v-2h14v2z"/>
</vector>
app/src/main/res/drawable/ic_plus.xml 0 → 100644
View file @ 2ccd44f5
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<vector xmlns:android="http://schemas.android.com/apk/res/android"
android:width="24dp"
android:height="24dp"
android:viewportWidth="24"
android:viewportHeight="24">
<path
android:fillColor="#FF000000"
android:pathData="M19,13h-6v6h-2v-6H5v-2h6V5h2v6h6v2z"/>
</vector>
app/src/main/res/layout/activity_main.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<androidx.coordinatorlayout.widget.CoordinatorLayout
xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:tools="http://schemas.android.com/tools"
xmlns:app="http://schemas.android.com/apk/res-auto"
android:background="@android:color/transparent"
android:layout_width="match_parent"
android:layout_height="match_parent">
<RelativeLayout
android:layout_width="match_parent"
android:layout_height="match_parent"
android:orientation="vertical">
<androidx.fragment.app.FragmentContainerView
android:id="@+id/fragment_container"
android:name="androidx.navigation.fragment.NavHostFragment"
android:layout_width="match_parent"
android:layout_height="match_parent"
android:background="@android:color/transparent"
android:keepScreenOn="true"
app:defaultNavHost="true"
app:navGraph="@navigation/nav_graph"
android:layout_marginTop="?android:attr/actionBarSize"
tools:context=".MainActivity"/>
<androidx.appcompat.widget.Toolbar
android:id="@+id/toolbar"
android:layout_width="match_parent"
android:layout_height="?attr/actionBarSize"
android:layout_alignParentTop="true"
android:background="@color/toolbar_background">
<ImageView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:src="@drawable/tfl_logo" />
</androidx.appcompat.widget.Toolbar>
</RelativeLayout>
</androidx.coordinatorlayout.widget.CoordinatorLayout>
\ No newline at end of file
app/src/main/res/layout/fragment_camera.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<androidx.coordinatorlayout.widget.CoordinatorLayout xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:app="http://schemas.android.com/apk/res-auto"
android:id="@+id/camera_container"
android:layout_width="match_parent"
android:layout_height="match_parent">
<androidx.camera.view.PreviewView
android:id="@+id/view_finder"
android:layout_width="match_parent"
android:layout_height="match_parent"
app:scaleType="fillStart"/>
<org.tensorflow.lite.examples.objectdetection.OverlayView
android:id="@+id/overlay"
android:layout_height="match_parent"
android:layout_width="match_parent" />
<include
android:id="@+id/bottom_sheet_layout"
layout="@layout/info_bottom_sheet" />
</androidx.coordinatorlayout.widget.CoordinatorLayout>
\ No newline at end of file
app/src/main/res/layout/info_bottom_sheet.xml 0 → 100644
View file @ 2ccd44f5
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<androidx.core.widget.NestedScrollView xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:app="http://schemas.android.com/apk/res-auto"
android:id="@+id/bottom_sheet_layout"
android:layout_width="match_parent"
android:layout_gravity="center_horizontal"
android:background="@color/bottom_sheet_background"
android:gravity="center_horizontal"
android:orientation="vertical"
android:padding="@dimen/bottom_sheet_padding"
android:layout_height="wrap_content"
app:behavior_hideable="false"
app:behavior_peekHeight="@dimen/bottom_sheet_peek_height"
app:layout_behavior="com.google.android.material.bottomsheet.BottomSheetBehavior">
<androidx.appcompat.widget.LinearLayoutCompat
android:orientation="vertical"
android:layout_width="match_parent"
android:layout_height="match_parent">
<!-- Chevron indicating that the bottom sheet is expandable -->
<RelativeLayout
android:layout_height="wrap_content"
android:layout_width="match_parent">
<ImageView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_centerHorizontal="true"
android:layout_centerVertical="true"
android:src="@drawable/icn_chevron_up"
android:contentDescription="@string/alt_bottom_sheet_chevron" />
</RelativeLayout>
<!-- Inference time row -->
<androidx.appcompat.widget.LinearLayoutCompat
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin"
android:orientation="horizontal">
<TextView
android:id="@+id/inference_time_label"
android:text="@string/label_interence_time"
android:textColor="@color/bottom_sheet_text_color"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_centerVertical="true"
android:textSize="@dimen/bottom_sheet_text_size" />
<TextView
android:id="@+id/inference_time_val"
android:text="0ms"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:gravity="end"
android:layout_centerVertical="true"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
</androidx.appcompat.widget.LinearLayoutCompat>
<!-- Active scan type row -->
<androidx.appcompat.widget.LinearLayoutCompat
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin"
android:orientation="horizontal">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="@string/label_scan_type"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
<TextView
android:id="@+id/scan_type_value"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:gravity="end"
android:text="@string/scan_type_people"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
</androidx.appcompat.widget.LinearLayoutCompat>
<!-- Number of people in the current frame -->
<androidx.appcompat.widget.LinearLayoutCompat
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin"
android:orientation="horizontal">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="@string/label_detected_count"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
<TextView
android:id="@+id/detected_count_value"
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:gravity="end"
android:text="0"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
</androidx.appcompat.widget.LinearLayoutCompat>
<!-- ML confidence threshold adjustment row -->
<RelativeLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin"
android:gravity="center"
android:orientation="horizontal">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="@string/label_confidence_threshold"
android:layout_centerVertical="true"
android:textSize="@dimen/bottom_sheet_text_size"
android:textColor="@color/bottom_sheet_text_color" />
<LinearLayout
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_alignParentRight="true"
android:gravity="center_vertical"
android:orientation="horizontal"
>
<androidx.appcompat.widget.AppCompatImageButton
android:id="@+id/threshold_minus"
android:layout_width="@dimen/bottom_sheet_control_btn_size"
android:layout_height="@dimen/bottom_sheet_control_btn_size"
android:contentDescription="@string/alt_bottom_sheet_threshold_button_minus"
android:src="@drawable/ic_minus" />
<TextView
android:id="@+id/threshold_value"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_marginLeft="@dimen/bottom_sheet_control_text_side_margin"
android:layout_marginRight="@dimen/bottom_sheet_control_text_side_margin"
android:gravity="center"
android:minEms="@integer/bottom_sheet_control_text_min_ems"
android:text="0.50"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
<androidx.appcompat.widget.AppCompatImageButton
android:id="@+id/threshold_plus"
android:layout_width="@dimen/bottom_sheet_control_btn_size"
android:layout_height="@dimen/bottom_sheet_control_btn_size"
android:contentDescription="@string/alt_bottom_sheet_threshold_button_plus"
android:src="@drawable/ic_plus" />
</LinearLayout>
</RelativeLayout>
<!-- ML max results adjustment row -->
<RelativeLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin"
android:gravity="center"
android:orientation="horizontal">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="@string/label_max_results"
android:textSize="@dimen/bottom_sheet_text_size"
android:layout_centerVertical="true"
android:textColor="@color/bottom_sheet_text_color" />
<LinearLayout
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_alignParentRight="true"
android:gravity="center"
android:orientation="horizontal">
<androidx.appcompat.widget.AppCompatImageButton
android:id="@+id/max_results_minus"
android:layout_width="@dimen/bottom_sheet_control_btn_size"
android:layout_height="@dimen/bottom_sheet_control_btn_size"
android:contentDescription="@string/alt_bottom_sheet_max_results_button_minus"
android:src="@drawable/ic_minus" />
<TextView
android:id="@+id/max_results_value"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:minEms="@integer/bottom_sheet_control_text_min_ems"
android:gravity="center"
android:layout_marginLeft="@dimen/bottom_sheet_control_text_side_margin"
android:layout_marginRight="@dimen/bottom_sheet_control_text_side_margin"
android:text="20"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
<androidx.appcompat.widget.AppCompatImageButton
android:id="@+id/max_results_plus"
android:layout_width="@dimen/bottom_sheet_control_btn_size"
android:layout_height="@dimen/bottom_sheet_control_btn_size"
android:contentDescription="@string/alt_bottom_sheet_max_results_button_plus"
android:src="@drawable/ic_plus" />
</LinearLayout>
</RelativeLayout>
<!-- Number of threads adjustment row -->
<RelativeLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin"
android:gravity="center"
android:orientation="horizontal">
<TextView
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:text="@string/label_threads"
android:layout_centerVertical="true"
android:textSize="@dimen/bottom_sheet_text_size"
android:textColor="@color/bottom_sheet_text_color" />
<LinearLayout
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_alignParentRight="true"
android:gravity="center_vertical"
android:orientation="horizontal"
>
<androidx.appcompat.widget.AppCompatImageButton
android:id="@+id/threads_minus"
android:layout_width="@dimen/bottom_sheet_control_btn_size"
android:layout_height="@dimen/bottom_sheet_control_btn_size"
android:contentDescription="@string/alt_bottom_sheet_thread_button_minus"
android:src="@drawable/ic_minus" />
<TextView
android:id="@+id/threads_value"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:layout_marginLeft="@dimen/bottom_sheet_control_text_side_margin"
android:layout_marginRight="@dimen/bottom_sheet_control_text_side_margin"
android:gravity="center"
android:minEms="@integer/bottom_sheet_control_text_min_ems"
android:text="2"
android:textColor="@color/bottom_sheet_text_color"
android:textSize="@dimen/bottom_sheet_text_size" />
<androidx.appcompat.widget.AppCompatImageButton
android:id="@+id/threads_plus"
android:layout_width="@dimen/bottom_sheet_control_btn_size"
android:layout_height="@dimen/bottom_sheet_control_btn_size"
android:contentDescription="@string/alt_bottom_sheet_thread_button_plus"
android:src="@drawable/ic_plus" />
</LinearLayout>
</RelativeLayout>
<!-- Delegate selection row -->
<RelativeLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin">
<TextView
android:text="@string/label_delegate"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:textSize="@dimen/bottom_sheet_text_size"
android:textColor="@color/bottom_sheet_text_color" />
<androidx.appcompat.widget.AppCompatSpinner
android:id="@+id/spinner_delegate"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:minWidth="@dimen/bottom_sheet_spinner_delegate_min_width"
android:spinnerMode="dropdown"
android:theme="@style/BottomSheetSpinnerItemStyle"
android:layout_alignParentRight="true"
android:entries="@array/delegate_spinner_titles"/>
</RelativeLayout>
<!-- Model selection row -->
<RelativeLayout
android:layout_width="match_parent"
android:layout_height="wrap_content"
android:layout_marginTop="@dimen/bottom_sheet_default_row_margin">
<TextView
android:text="@string/label_models"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:textSize="@dimen/bottom_sheet_text_size"
android:textColor="@color/bottom_sheet_text_color" />
<androidx.appcompat.widget.AppCompatSpinner
android:id="@+id/spinner_model"
android:layout_width="wrap_content"
android:layout_height="wrap_content"
android:minWidth="@dimen/bottom_sheet_spinner_model_min_width"
android:spinnerMode="dropdown"
android:theme="@style/BottomSheetSpinnerItemStyle"
android:layout_alignParentRight="true"
android:entries="@array/models_spinner_titles"/>
</RelativeLayout>
</androidx.appcompat.widget.LinearLayoutCompat>
</androidx.core.widget.NestedScrollView>
app/src/main/res/mipmap-anydpi-v26/ic_launcher.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ https://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<adaptive-icon xmlns:android="http://schemas.android.com/apk/res/android">
<background android:drawable="@color/ic_launcher_background"/>
<foreground android:drawable="@mipmap/ic_launcher_foreground"/>
</adaptive-icon>
\ No newline at end of file
app/src/main/res/mipmap-anydpi-v26/ic_launcher_round.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ https://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<adaptive-icon xmlns:android="http://schemas.android.com/apk/res/android">
<background android:drawable="@color/ic_launcher_background"/>
<foreground android:drawable="@mipmap/ic_launcher_foreground"/>
</adaptive-icon>
\ No newline at end of file
app/src/main/res/navigation/nav_graph.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<navigation
xmlns:android="http://schemas.android.com/apk/res/android"
xmlns:app="http://schemas.android.com/apk/res-auto"
android:id="@+id/nav_graph"
app:startDestination="@id/permissions_fragment">
<fragment
android:id="@+id/permissions_fragment"
android:name="org.tensorflow.lite.examples.objectdetection.fragments.PermissionsFragment"
android:label="PermissionsFragment" >
<action
android:id="@+id/action_permissions_to_camera"
app:destination="@id/camera_fragment"
app:popUpTo="@id/permissions_fragment"
app:popUpToInclusive="true" />
</fragment>
<fragment
android:id="@+id/camera_fragment"
android:name="org.tensorflow.lite.examples.objectdetection.fragments.CameraFragment"
android:label="CameraFragment" >
<action
android:id="@+id/action_camera_to_permissions"
app:destination="@id/permissions_fragment"
app:popUpTo="@id/camera_fragment"
app:popUpToInclusive="true"/>
</fragment>
</navigation>
\ No newline at end of file
app/src/main/res/values/colors.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<resources>
<!-- TF Branding Orange -->
<color name="bounding_box_color">#FF6F00</color>
<color name="toolbar_background">#EEEEEE</color>
<color name="ic_launcher_background">#FFFFFF</color>
<color name="bottom_sheet_background">#EEEEEE</color>
<color name="bottom_sheet_text_color">@android:color/black</color>
<color name="icActive">#FFFFFFFF</color>
<color name="icFocused">#DDFFFFFF</color>
<color name="icPressed">#AAFFFFFF</color>
</resources>
app/src/main/res/values/dimens.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<resources>
<dimen name="stroke_small">4dp</dimen>
<dimen name="round_button_medium">64dp</dimen>
<!-- Bottom Sheet -->
<dimen name="bottom_sheet_text_size">20sp</dimen>
<dimen name="bottom_sheet_padding">16dp</dimen>
<dimen name="bottom_sheet_peek_height">50dp</dimen>
<dimen name="bottom_sheet_default_row_margin">16dp</dimen>
<dimen name="bottom_sheet_control_btn_size">48dp</dimen>
<dimen name="bottom_sheet_control_text_side_margin">10dp</dimen>
<dimen name="bottom_sheet_spinner_delegate_min_width">160dp</dimen>
<dimen name="bottom_sheet_spinner_model_min_width">240dp</dimen>
<integer name="bottom_sheet_control_text_min_ems">3</integer>
</resources>
\ No newline at end of file
app/src/main/res/values/strings.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<resources>
<string name="app_name">TFLite Object Detection</string>
<string name="alt_bottom_sheet_chevron">Bottom sheet expandable indicator</string>
<string name="alt_bottom_sheet_max_results_button_minus">Decreasing maxium detected results
button</string>
<string name="alt_bottom_sheet_max_results_button_plus">Increasing maxium detected results
button</string>
<string name="alt_bottom_sheet_threshold_button_minus">Decreasing threshold of detected object
results button</string>
<string name="alt_bottom_sheet_threshold_button_plus">Increasing threshold of detected object
results button</string>
<string name="alt_bottom_sheet_thread_button_minus">Decrease the number of threads used</string>
<string name="alt_bottom_sheet_thread_button_plus">Increase the number of threads used</string>
<string name="label_interence_time">Inference Time</string>
<string name="label_fps">Frames per Second</string>
<string name="label_scan_type">Scan Type</string>
<string name="scan_type_people">People</string>
<string name="label_detected_count">People Detected</string>
<string name="label_confidence_threshold">Threshold</string>
<string name="label_max_results">Max Results</string>
<string name="label_threads">Number of Threads</string>
<string name="label_delegate">Delegate</string>
<string name="label_models">ML Model</string>
<string-array name="delegate_spinner_titles">
<item>CPU</item>
<item>GPU</item>
<item>NNAPI</item>
</string-array>
<string-array name="models_spinner_titles">
<item>MobileNet V1</item>
<item>EfficientDet Lite0</item>
<item>EfficientDet Lite1</item>
<item>EfficientDet Lite2</item>
</string-array>
</resources>
app/src/main/res/values/styles.xml 0 → 100644
View file @ 2ccd44f5
<?xml version="1.0" encoding="utf-8"?>
<!--
~ Copyright 2022 The TensorFlow Authors. All Rights Reserved.
~
~ Licensed under the Apache License, Version 2.0 (the "License");
~ you may not use this file except in compliance with the License.
~ You may obtain a copy of the License at
~
~ http://www.apache.org/licenses/LICENSE-2.0
~
~ Unless required by applicable law or agreed to in writing, software
~ distributed under the License is distributed on an "AS IS" BASIS,
~ WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
~ See the License for the specific language governing permissions and
~ limitations under the License.
-->
<resources>
<!-- Base application theme. -->
<style name="AppTheme" parent="Theme.AppCompat.Light.NoActionBar">
</style>
<style name="BottomSheetSpinnerItemStyle" parent="Widget.AppCompat.Light.DropDownItem.Spinner">
<item name="android:textSize">@dimen/bottom_sheet_text_size</item>
</style>
</resources>
build.gradle 0 → 100644
View file @ 2ccd44f5
// Top-level build file where you can add configuration options common to all sub-projects/modules.
buildscript {
// Top-level variables used for versioning
ext.kotlin_version = '1.9.22'
ext.java_version = JavaVersion.VERSION_17
repositories {
google()
mavenCentral()
}
dependencies {
classpath 'com.android.tools.build:gradle:8.2.2'
classpath "org.jetbrains.kotlin:kotlin-gradle-plugin:$kotlin_version"
classpath 'androidx.navigation:navigation-safe-args-gradle-plugin:2.7.7'
classpath 'de.undercouch:gradle-download-task:5.0.5'
// NOTE: Do not place your application dependencies here; they belong
// in the individual module build.gradle files
}
}
allprojects {
repositories {
google()
mavenCentral()
}
}
task clean(type: Delete) {
delete rootProject.buildDir
}
gradle.properties 0 → 100644
View file @ 2ccd44f5
org.gradle.jvmargs=-Xmx1536m
android.enableJetifier=true
android.useAndroidX=true
\ No newline at end of file
gradle/wrapper/gradle-wrapper.properties 0 → 100644
View file @ 2ccd44f5
distributionBase=GRADLE_USER_HOME
distributionPath=wrapper/dists
distributionUrl=https\://services.gradle.org/distributions/gradle-8.5-bin.zip
networkTimeout=10000
retries=0
retryBackOffMs=500
validateDistributionUrl=true
zipStoreBase=GRADLE_USER_HOME
zipStorePath=wrapper/dists
gradlew 0 → 100644
View file @ 2ccd44f5
#!/bin/sh
#
# Copyright © 2015 the original authors.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# https://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
#
# SPDX-License-Identifier: Apache-2.0
#
##############################################################################
#
# gradlew start up script for POSIX generated by Gradle.
#
# Important for running:
#
# (1) You need a POSIX-compliant shell to run this script. If your /bin/sh is
# noncompliant, but you have some other compliant shell such as ksh or
# bash, then to run this script, type that shell name before the whole
# command line, like:
#
# ksh gradlew
#
# Busybox and similar reduced shells will NOT work, because this script
# requires all of these POSIX shell features:
# * functions;
# * expansions «$var», «${var}», «${var:-default}», «${var+SET}»,
# «${var#prefix}», «${var%suffix}», and «$( cmd )»;
# * compound commands having a testable exit status, especially «case»;
# * various built-in commands including «command», «set», and «ulimit».
#
# Important for patching:
#
# (2) This script targets any POSIX shell, so it avoids extensions provided
# by Bash, Ksh, etc; in particular arrays are avoided.
#
# The "traditional" practice of packing multiple parameters into a
# space-separated string is a well documented source of bugs and security
# problems, so this is (mostly) avoided, by progressively accumulating
# options in "$@", and eventually passing that to Java.
#
# Where the inherited environment variables (DEFAULT_JVM_OPTS, JAVA_OPTS,
# and GRADLE_OPTS) rely on word-splitting, this is performed explicitly;
# see the in-line comments for details.
#
# There are tweaks for specific operating systems such as AIX, CygWin,
# Darwin, MinGW, and NonStop.
#
# (3) This script is generated from the Groovy template
# https://github.com/gradle/gradle/blob/3d91ce3b8caaf77ad09f381f43615b715b53f72c/platforms/jvm/plugins-application/src/main/resources/org/gradle/api/internal/plugins/unixStartScript.txt
# within the Gradle project.
#
# You can find Gradle at https://github.com/gradle/gradle/.
#
##############################################################################
# Attempt to set APP_HOME
# Resolve links: $0 may be a link
app_path=$0
# Need this for daisy-chained symlinks.
while
APP_HOME=${app_path%"${app_path##*/}"} # leaves a trailing /; empty if no leading path
[ -h "$app_path" ]
do
ls=$( ls -ld "$app_path" )
link=${ls#*' -> '}
case $link in #(
/*) app_path=$link ;; #(
*) app_path=$APP_HOME$link ;;
esac
done
# This is normally unused
# shellcheck disable=SC2034
APP_BASE_NAME=${0##*/}
# Discard cd standard output in case $CDPATH is set (https://github.com/gradle/gradle/issues/25036)
APP_HOME=$( cd -P "${APP_HOME:-./}" > /dev/null && printf '%s\n' "$PWD" ) || exit
# Use the maximum available, or set MAX_FD != -1 to use that value.
MAX_FD=maximum
warn () {
echo "$*"
} >&2
die () {
echo
echo "$*"
echo
exit 1
} >&2
# OS specific support (must be 'true' or 'false').
cygwin=false
msys=false
darwin=false
nonstop=false
case "$( uname )" in #(
CYGWIN* ) cygwin=true ;; #(
Darwin* ) darwin=true ;; #(
MSYS* | MINGW* ) msys=true ;; #(
NONSTOP* ) nonstop=true ;;
esac
# Determine the Java command to use to start the JVM.
if [ -n "$JAVA_HOME" ] ; then
if [ -x "$JAVA_HOME/jre/sh/java" ] ; then
# IBM's JDK on AIX uses strange locations for the executables
JAVACMD=$JAVA_HOME/jre/sh/java
else
JAVACMD=$JAVA_HOME/bin/java
fi
if [ ! -x "$JAVACMD" ] ; then
die "ERROR: JAVA_HOME is set to an invalid directory: $JAVA_HOME
Please set the JAVA_HOME variable in your environment to match the
location of your Java installation."
fi
else
JAVACMD=java
if ! command -v java >/dev/null 2>&1
then
die "ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH.
Please set the JAVA_HOME variable in your environment to match the
location of your Java installation."
fi
fi
# Increase the maximum file descriptors if we can.
if ! "$cygwin" && ! "$darwin" && ! "$nonstop" ; then
case $MAX_FD in #(
max*)
# In POSIX sh, ulimit -H is undefined. That's why the result is checked to see if it worked.
# shellcheck disable=SC2039,SC3045
MAX_FD=$( ulimit -H -n ) ||
warn "Could not query maximum file descriptor limit"
esac
case $MAX_FD in #(
'' | soft) :;; #(
*)
# In POSIX sh, ulimit -n is undefined. That's why the result is checked to see if it worked.
# shellcheck disable=SC2039,SC3045
ulimit -n "$MAX_FD" ||
warn "Could not set maximum file descriptor limit to $MAX_FD"
esac
fi
# Collect all arguments for the java command, stacking in reverse order:
# * args from the command line
# * the main class name
# * -classpath
# * -D...appname settings
# * --module-path (only if needed)
# * DEFAULT_JVM_OPTS, JAVA_OPTS, and GRADLE_OPTS environment variables.
# For Cygwin or MSYS, switch paths to Windows format before running java
if "$cygwin" || "$msys" ; then
APP_HOME=$( cygpath --path --mixed "$APP_HOME" )
JAVACMD=$( cygpath --unix "$JAVACMD" )
# Now convert the arguments - kludge to limit ourselves to /bin/sh
for arg do
if
case $arg in #(
-*) false ;; # don't mess with options #(
/?*) t=${arg#/} t=/${t%%/*} # looks like a POSIX filepath
[ -e "$t" ] ;; #(
*) false ;;
esac
then
arg=$( cygpath --path --ignore --mixed "$arg" )
fi
# Roll the args list around exactly as many times as the number of
# args, so each arg winds up back in the position where it started, but
# possibly modified.
#
# NB: a `for` loop captures its iteration list before it begins, so
# changing the positional parameters here affects neither the number of
# iterations, nor the values presented in `arg`.
shift # remove old arg
set -- "$@" "$arg" # push replacement arg
done
fi
# Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
DEFAULT_JVM_OPTS='"-Xmx64m" "-Xms64m"'
# Collect all arguments for the java command:
# * DEFAULT_JVM_OPTS, JAVA_OPTS, and optsEnvironmentVar are not allowed to contain shell fragments,
# and any embedded shellness will be escaped.
# * For example: A user cannot expect ${Hostname} to be expanded, as it is an environment variable and will be
# treated as '${Hostname}' itself on the command line.
set -- \
"-Dorg.gradle.appname=$APP_BASE_NAME" \
-jar "$APP_HOME/gradle/wrapper/gradle-wrapper.jar" \
"$@"
# Stop when "xargs" is not available.
if ! command -v xargs >/dev/null 2>&1
then
die "xargs is not available"
fi
# Use "xargs" to parse quoted args.
#
# With -n1 it outputs one arg per line, with the quotes and backslashes removed.
#
# In Bash we could simply go:
#
# readarray ARGS < <( xargs -n1 <<<"$var" ) &&
# set -- "${ARGS[@]}" "$@"
#
# but POSIX shell has neither arrays nor command substitution, so instead we
# post-process each arg (as a line of input to sed) to backslash-escape any
# character that might be a shell metacharacter, then use eval to reverse
# that process (while maintaining the separation between arguments), and wrap
# the whole thing up as a single "set" statement.
#
# This will of course break if any of these variables contains a newline or
# an unmatched quote.
#
eval "set -- $(
printf '%s\n' "$DEFAULT_JVM_OPTS $JAVA_OPTS $GRADLE_OPTS" |
xargs -n1 |
sed ' s~[^-[:alnum:]+,./:=@_]~\\&~g; ' |
tr '\n' ' '
)" '"$@"'
exec "$JAVACMD" "$@"
gradlew.bat 0 → 100644
View file @ 2ccd44f5
@rem
@rem Copyright 2015 the original author or authors.
@rem
@rem Licensed under the Apache License, Version 2.0 (the "License");
@rem you may not use this file except in compliance with the License.
@rem You may obtain a copy of the License at
@rem
@rem https://www.apache.org/licenses/LICENSE-2.0
@rem
@rem Unless required by applicable law or agreed to in writing, software
@rem distributed under the License is distributed on an "AS IS" BASIS,
@rem WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@rem See the License for the specific language governing permissions and
@rem limitations under the License.
@rem
@rem SPDX-License-Identifier: Apache-2.0
@rem
@if "%DEBUG%"=="" @echo off
@rem ##########################################################################
@rem
@rem gradlew startup script for Windows
@rem
@rem ##########################################################################
@rem Set local scope for the variables, and ensure extensions are enabled
setlocal EnableExtensions
set DIRNAME=%~dp0
if "%DIRNAME%"=="" set DIRNAME=.
@rem This is normally unused
set APP_BASE_NAME=%~n0
set APP_HOME=%DIRNAME%
@rem Resolve any "." and ".." in APP_HOME to make it shorter.
for %%i in ("%APP_HOME%") do set APP_HOME=%%~fi
@rem Add default JVM options here. You can also use JAVA_OPTS and GRADLE_OPTS to pass JVM options to this script.
set DEFAULT_JVM_OPTS="-Xmx64m" "-Xms64m"
@rem Find java.exe
if defined JAVA_HOME goto findJavaFromJavaHome
set JAVA_EXE=java.exe
%JAVA_EXE% -version >NUL 2>&1
if %ERRORLEVEL% equ 0 goto execute
echo. 1>&2
echo ERROR: JAVA_HOME is not set and no 'java' command could be found in your PATH. 1>&2
echo. 1>&2
echo Please set the JAVA_HOME variable in your environment to match the 1>&2
echo location of your Java installation. 1>&2
"%COMSPEC%" /c exit 1
:findJavaFromJavaHome
set JAVA_HOME=%JAVA_HOME:"=%
set JAVA_EXE=%JAVA_HOME%/bin/java.exe
if exist "%JAVA_EXE%" goto execute
echo. 1>&2
echo ERROR: JAVA_HOME is set to an invalid directory: %JAVA_HOME% 1>&2
echo. 1>&2
echo Please set the JAVA_HOME variable in your environment to match the 1>&2
echo location of your Java installation. 1>&2
"%COMSPEC%" /c exit 1
:execute
@rem Setup the command line
@rem Execute gradlew
@rem endlocal doesn't take effect until after the line is parsed and variables are expanded
@rem which allows us to clear the local environment before executing the java command
endlocal & "%JAVA_EXE%" %DEFAULT_JVM_OPTS% %JAVA_OPTS% %GRADLE_OPTS% "-Dorg.gradle.appname=%APP_BASE_NAME%" -jar "%APP_HOME%\gradle\wrapper\gradle-wrapper.jar" %* & call :exitWithErrorLevel
:exitWithErrorLevel
@rem Use "%COMSPEC%" /c exit to allow operators to work properly in scripts
"%COMSPEC%" /c exit %ERRORLEVEL%
settings.gradle 0 → 100644
View file @ 2ccd44f5
include ':app'
\ No newline at end of file
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment