شما میتوانید از کیت ML برای تشخیص و ردیابی اشیاء در فریمهای ویدیویی متوالی استفاده کنید.
وقتی تصویری را به کیت ML ارسال میکنید، کیت حداکثر پنج شیء را در تصویر به همراه موقعیت هر شیء در تصویر تشخیص میدهد. هنگام تشخیص اشیاء در جریانهای ویدیویی، هر شیء یک شناسه منحصر به فرد دارد که میتوانید از آن برای ردیابی شیء از فریمی به فریم دیگر استفاده کنید. همچنین میتوانید به صورت اختیاری طبقهبندی کلی اشیاء را فعال کنید که اشیاء را با توضیحات کلی دستهبندی میکند.
امتحانش کن.
- برای مشاهدهی نحوهی استفاده از این API، با برنامهی نمونه کار کنید.
- برای پیادهسازی کامل این API ، به برنامهی نمایشی Material Design مراجعه کنید.
قبل از اینکه شروع کنی
- در فایل
build.gradleسطح پروژه خود، مطمئن شوید که مخزن Maven گوگل را هم در بخشهایbuildscriptو همallprojectsخود وارد کردهاید. - وابستگیهای کتابخانههای اندروید ML Kit را به فایل gradle سطح app ماژول خود که معمولاً
app/build.gradleاست، اضافه کنید:dependencies { // ... implementation 'com.google.mlkit:object-detection:17.0.2' }
۱. آشکارساز شیء را پیکربندی کنید
برای شناسایی و ردیابی اشیاء، ابتدا یک نمونه از ObjectDetector ایجاد کنید و به صورت اختیاری هر تنظیمات آشکارسازی را که میخواهید از حالت پیشفرض تغییر دهید، مشخص کنید.
آشکارساز شیء را برای مورد استفاده خود با یک شیء
ObjectDetectorOptionsپیکربندی کنید. میتوانید تنظیمات زیر را تغییر دهید:تنظیمات آشکارساز شیء حالت تشخیص STREAM_MODE(پیشفرض) |SINGLE_IMAGE_MODEدر حالت
STREAM_MODE(پیشفرض)، آشکارساز شیء با تأخیر کم اجرا میشود، اما ممکن است در چند فراخوانی اول آشکارساز، نتایج ناقصی (مانند کادرهای محدودکننده نامشخص یا برچسبهای دستهبندی) تولید کند. همچنین، درSTREAM_MODE، آشکارساز شناسههای ردیابی را به اشیاء اختصاص میدهد که میتوانید از آنها برای ردیابی اشیاء در فریمها استفاده کنید. از این حالت زمانی استفاده کنید که میخواهید اشیاء را ردیابی کنید یا زمانی که تأخیر کم مهم است، مانند پردازش جریانهای ویدیویی در زمان واقعی.در
SINGLE_IMAGE_MODE، آشکارساز شیء نتیجه را پس از تعیین محدوده شیء برمیگرداند. اگر طبقهبندی را نیز فعال کنید، نتیجه را پس از در دسترس بودن محدوده و برچسب دستهبندی برمیگرداند. در نتیجه، تأخیر تشخیص به طور بالقوه بیشتر است. همچنین، درSINGLE_IMAGE_MODE، شناسههای ردیابی اختصاص داده نمیشوند. اگر تأخیر حیاتی نیست و نمیخواهید با نتایج جزئی سر و کار داشته باشید، از این حالت استفاده کنید.تشخیص و ردیابی چندین شیء false(پیشفرض) |trueاینکه آیا تا پنج شیء شناسایی و ردیابی شود یا فقط برجستهترین شیء (پیشفرض).
طبقهبندی اشیاء false(پیشفرض) |trueاینکه آیا اشیاء شناساییشده در دستههای کلی طبقهبندی شوند یا خیر. در صورت فعال بودن، آشکارساز شیء، اشیاء را در دستههای زیر طبقهبندی میکند: کالاهای مد، غذا، کالاهای خانگی، مکانها و گیاهان.
API تشخیص و ردیابی اشیا برای این دو مورد استفاده اصلی بهینه شده است:
- تشخیص و ردیابی زنده برجستهترین شیء در منظرهیاب دوربین.
- تشخیص چندین شیء از یک تصویر ثابت
برای پیکربندی API برای این موارد استفاده:
کاتلین
// Live detection and tracking val options = ObjectDetectorOptions.Builder() .setDetectorMode(ObjectDetectorOptions.STREAM_MODE) .enableClassification() // Optional .build() // Multiple object detection in static images val options = ObjectDetectorOptions.Builder() .setDetectorMode(ObjectDetectorOptions.SINGLE_IMAGE_MODE) .enableMultipleObjects() .enableClassification() // Optional .build()
جاوا
// Live detection and tracking ObjectDetectorOptions options = new ObjectDetectorOptions.Builder() .setDetectorMode(ObjectDetectorOptions.STREAM_MODE) .enableClassification() // Optional .build(); // Multiple object detection in static images ObjectDetectorOptions options = new ObjectDetectorOptions.Builder() .setDetectorMode(ObjectDetectorOptions.SINGLE_IMAGE_MODE) .enableMultipleObjects() .enableClassification() // Optional .build();
یک نمونه از
ObjectDetectorدریافت کنید:کاتلین
val objectDetector = ObjectDetection.getClient(options)
جاوا
ObjectDetector objectDetector = ObjectDetection.getClient(options);
۲. تصویر ورودی را آماده کنید
برای شناسایی و ردیابی اشیاء، تصاویر را به متدprocess() از نمونه ObjectDetector ارسال کنید. تشخیصدهندهی شیء مستقیماً از یک Bitmap ، NV21 ByteBuffer یا یک YUV_420_888 media.Image اجرا میشود. اگر به یکی از این منابع دسترسی مستقیم دارید، ساخت یک InputImage از آنها توصیه میشود. اگر یک InputImage از منابع دیگر بسازید، ما تبدیل را به صورت داخلی برای شما انجام خواهیم داد و ممکن است کارایی کمتری داشته باشد.
برای هر فریم از ویدیو یا تصویر در یک دنباله، موارد زیر را انجام دهید:
شما میتوانید یک شیء InputImage را از منابع مختلفی ایجاد کنید که هر کدام در زیر توضیح داده شدهاند.
استفاده از یک media.Image
برای ایجاد یک شیء InputImage از یک شیء media.Image ، مانند زمانی که از دوربین یک دستگاه تصویر میگیرید، شیء media.Image و چرخش تصویر را به InputImage.fromMediaImage() ارسال کنید.
اگر از کتابخانه CameraX استفاده میکنید، کلاسهای OnImageCapturedListener و ImageAnalysis.Analyzer مقدار چرخش را برای شما محاسبه میکنند.
کاتلین
private class YourImageAnalyzer : ImageAnalysis.Analyzer { override fun analyze(imageProxy: ImageProxy) { val mediaImage = imageProxy.image if (mediaImage != null) { val image = InputImage.fromMediaImage(mediaImage, imageProxy.imageInfo.rotationDegrees) // Pass image to an ML Kit Vision API // ... } } }
جاوا
private class YourAnalyzer implements ImageAnalysis.Analyzer { @Override public void analyze(ImageProxy imageProxy) { Image mediaImage = imageProxy.getImage(); if (mediaImage != null) { InputImage image = InputImage.fromMediaImage(mediaImage, imageProxy.getImageInfo().getRotationDegrees()); // Pass image to an ML Kit Vision API // ... } } }
اگر از کتابخانه دوربینی که درجه چرخش تصویر را به شما بدهد استفاده نمیکنید، میتوانید آن را از درجه چرخش دستگاه و جهت سنسور دوربین در دستگاه محاسبه کنید:
کاتلین
private val ORIENTATIONS = SparseIntArray() init { ORIENTATIONS.append(Surface.ROTATION_0, 0) ORIENTATIONS.append(Surface.ROTATION_90, 90) ORIENTATIONS.append(Surface.ROTATION_180, 180) ORIENTATIONS.append(Surface.ROTATION_270, 270) } /** * Get the angle by which an image must be rotated given the device's current * orientation. */ @RequiresApi(api = Build.VERSION_CODES.LOLLIPOP) @Throws(CameraAccessException::class) private fun getRotationCompensation(cameraId: String, activity: Activity, isFrontFacing: Boolean): Int { // Get the device's current rotation relative to its "native" orientation. // Then, from the ORIENTATIONS table, look up the angle the image must be // rotated to compensate for the device's rotation. val deviceRotation = activity.windowManager.defaultDisplay.rotation var rotationCompensation = ORIENTATIONS.get(deviceRotation) // Get the device's sensor orientation. val cameraManager = activity.getSystemService(CAMERA_SERVICE) as CameraManager val sensorOrientation = cameraManager .getCameraCharacteristics(cameraId) .get(CameraCharacteristics.SENSOR_ORIENTATION)!! if (isFrontFacing) { rotationCompensation = (sensorOrientation + rotationCompensation) % 360 } else { // back-facing rotationCompensation = (sensorOrientation - rotationCompensation + 360) % 360 } return rotationCompensation }
جاوا
private static final SparseIntArray ORIENTATIONS = new SparseIntArray(); static { ORIENTATIONS.append(Surface.ROTATION_0, 0); ORIENTATIONS.append(Surface.ROTATION_90, 90); ORIENTATIONS.append(Surface.ROTATION_180, 180); ORIENTATIONS.append(Surface.ROTATION_270, 270); } /** * Get the angle by which an image must be rotated given the device's current * orientation. */ @RequiresApi(api = Build.VERSION_CODES.LOLLIPOP) private int getRotationCompensation(String cameraId, Activity activity, boolean isFrontFacing) throws CameraAccessException { // Get the device's current rotation relative to its "native" orientation. // Then, from the ORIENTATIONS table, look up the angle the image must be // rotated to compensate for the device's rotation. int deviceRotation = activity.getWindowManager().getDefaultDisplay().getRotation(); int rotationCompensation = ORIENTATIONS.get(deviceRotation); // Get the device's sensor orientation. CameraManager cameraManager = (CameraManager) activity.getSystemService(CAMERA_SERVICE); int sensorOrientation = cameraManager .getCameraCharacteristics(cameraId) .get(CameraCharacteristics.SENSOR_ORIENTATION); if (isFrontFacing) { rotationCompensation = (sensorOrientation + rotationCompensation) % 360; } else { // back-facing rotationCompensation = (sensorOrientation - rotationCompensation + 360) % 360; } return rotationCompensation; }
سپس، شیء media.Image و مقدار درجه چرخش را به InputImage.fromMediaImage() ارسال کنید:
کاتلین
val image = InputImage.fromMediaImage(mediaImage, rotation)
Java
InputImage image = InputImage.fromMediaImage(mediaImage, rotation);
استفاده از یک URI فایل
برای ایجاد یک شیء InputImage از یک URI فایل، متن برنامه و URI فایل را به InputImage.fromFilePath() ارسال کنید. این زمانی مفید است که از یک ACTION_GET_CONTENT برای وادار کردن کاربر به انتخاب یک تصویر از برنامه گالری خود استفاده میکنید.
کاتلین
val image: InputImage try { image = InputImage.fromFilePath(context, uri) } catch (e: IOException) { e.printStackTrace() }
Java
InputImage image; try { image = InputImage.fromFilePath(context, uri); } catch (IOException e) { e.printStackTrace(); }
استفاده از ByteBuffer یا ByteArray
برای ایجاد یک شیء InputImage از یک ByteBuffer یا یک ByteArray ، ابتدا درجه چرخش تصویر را همانطور که قبلاً برای ورودی media.Image توضیح داده شد، محاسبه کنید. سپس، شیء InputImage را با بافر یا آرایه، به همراه ارتفاع، عرض، فرمت کدگذاری رنگ و درجه چرخش تصویر ایجاد کنید:
کاتلین
val image = InputImage.fromByteBuffer( byteBuffer, /* image width */ 480, /* image height */ 360, rotationDegrees, InputImage.IMAGE_FORMAT_NV21 // or IMAGE_FORMAT_YV12 ) // Or: val image = InputImage.fromByteArray( byteArray, /* image width */ 480, /* image height */ 360, rotationDegrees, InputImage.IMAGE_FORMAT_NV21 // or IMAGE_FORMAT_YV12 )
جاوا
InputImage image = InputImage.fromByteBuffer(byteBuffer, /* image width */ 480, /* image height */ 360, rotationDegrees, InputImage.IMAGE_FORMAT_NV21 // or IMAGE_FORMAT_YV12 ); // Or: InputImage image = InputImage.fromByteArray( byteArray, /* image width */480, /* image height */360, rotation, InputImage.IMAGE_FORMAT_NV21 // or IMAGE_FORMAT_YV12 );
استفاده از Bitmap
برای ایجاد یک شیء InputImage از یک شیء Bitmap ، تعریف زیر را انجام دهید:
کاتلین
val image = InputImage.fromBitmap(bitmap, 0)
Java
InputImage image = InputImage.fromBitmap(bitmap, rotationDegree);
تصویر توسط یک شیء Bitmap به همراه درجه چرخش نمایش داده میشود.
۳. تصویر را پردازش کنید
تصویر را به متدprocess() ارسال کنید: کاتلین
objectDetector.process(image) .addOnSuccessListener { detectedObjects -> // Task completed successfully // ... } .addOnFailureListener { e -> // Task failed with an exception // ... }
جاوا
objectDetector.process(image) .addOnSuccessListener( new OnSuccessListener<List<DetectedObject>>() { @Override public void onSuccess(List<DetectedObject> detectedObjects) { // Task completed successfully // ... } }) .addOnFailureListener( new OnFailureListener() { @Override public void onFailure(@NonNull Exception e) { // Task failed with an exception // ... } });
۴. اطلاعات مربوط به اشیاء شناسایی شده را دریافت کنید
اگر فراخوانی process() موفقیتآمیز باشد، فهرستی از DetectedObject ها به شنوندهی موفقیت ارسال میشود.
هر DetectedObject شامل ویژگیهای زیر است:
| جعبه محدود کننده | یک Rect که موقعیت شیء را در تصویر نشان میدهد. | ||||||
| شناسه ردیابی | یک عدد صحیح که شیء را در تصاویر مشخص میکند. در SINGLE_IMAGE_MODE مقدار null دارد. | ||||||
| برچسبها |
|
کاتلین
for (detectedObject in detectedObjects) { val boundingBox = detectedObject.boundingBox val trackingId = detectedObject.trackingId for (label in detectedObject.labels) { val text = label.text if (PredefinedCategory.FOOD == text) { ... } val index = label.index if (PredefinedCategory.FOOD_INDEX == index) { ... } val confidence = label.confidence } }
جاوا
// The list of detected objects contains one item if multiple // object detection wasn't enabled. for (DetectedObject detectedObject : detectedObjects) { Rect boundingBox = detectedObject.getBoundingBox(); Integer trackingId = detectedObject.getTrackingId(); for (Label label : detectedObject.getLabels()) { String text = label.getText(); if (PredefinedCategory.FOOD.equals(text)) { ... } int index = label.getIndex(); if (PredefinedCategory.FOOD_INDEX == index) { ... } float confidence = label.getConfidence(); } }
تضمین یک تجربه کاربری عالی
برای بهترین تجربه کاربری، این دستورالعملها را در برنامه خود دنبال کنید:
- تشخیص موفقیتآمیز اشیاء به پیچیدگی بصری شیء بستگی دارد. برای تشخیص، اشیاء با تعداد کمی از ویژگیهای بصری ممکن است نیاز داشته باشند که بخش بیشتری از تصویر را اشغال کنند. شما باید کاربران را در مورد دریافت ورودی که با نوع اشیاء مورد نظر شما مطابقت دارد، راهنمایی کنید.
- وقتی از طبقهبندی استفاده میکنید، اگر میخواهید اشیایی را که به طور کامل در دستهبندیهای پشتیبانیشده قرار نمیگیرند، شناسایی کنید، برای اشیای ناشناخته، روشهای خاصی را پیادهسازی کنید.
همچنین، اپلیکیشن ویترین ML Kit Material Design و مجموعه ویژگیهای Material Design Patterns for machine-powered را بررسی کنید.
بهبود عملکرد
اگر میخواهید از تشخیص اشیا در یک برنامهی بلادرنگ استفاده کنید، برای دستیابی به بهترین نرخ فریم، این دستورالعملها را دنبال کنید:
وقتی از حالت استریمینگ در یک برنامهی بلادرنگ استفاده میکنید، از تشخیص چند شیء استفاده نکنید، زیرا اکثر دستگاهها قادر به تولید فریمریت کافی نخواهند بود.
اگر به دستهبندی نیازی ندارید، آن را غیرفعال کنید.
- اگر از API
Cameraیاcamera2استفاده میکنید، فراخوانیهای throttle به آشکارساز انجام میشود. اگر در حین اجرای آشکارساز، یک فریم ویدیویی جدید در دسترس قرار گرفت، فریم را حذف کنید. برای مثال، به کلاسVisionProcessorBaseدر برنامه نمونه شروع سریع مراجعه کنید. - اگر از API
CameraXاستفاده میکنید، مطمئن شوید که استراتژی فشار معکوس (backpressure strategy) روی مقدار پیشفرض خودImageAnalysis.STRATEGY_KEEP_ONLY_LATESTتنظیم شده است. این تضمین میکند که فقط یک تصویر در هر زمان برای تجزیه و تحلیل تحویل داده میشود. اگر تصاویر بیشتری هنگام مشغول بودن تحلیلگر تولید شوند، به طور خودکار حذف میشوند و برای تحویل در صف قرار نمیگیرند. پس از بسته شدن تصویر در حال تجزیه و تحلیل با فراخوانی ImageProxy.close()، آخرین تصویر بعدی تحویل داده میشود. - اگر از خروجی آشکارساز برای همپوشانی گرافیک روی تصویر ورودی استفاده میکنید، ابتدا نتیجه را از کیت ML دریافت کنید، سپس تصویر و همپوشانی را در یک مرحله رندر کنید. این کار فقط یک بار برای هر فریم ورودی روی سطح نمایشگر رندر میشود. برای مثال به کلاسهای
CameraSourcePreviewوGraphicOverlayدر برنامه نمونه شروع سریع مراجعه کنید. - اگر از API دوربین ۲ استفاده میکنید، تصاویر را با فرمت
ImageFormat.YUV_420_888ضبط کنید. اگر از API دوربین قدیمیتر استفاده میکنید، تصاویر را با فرمتImageFormat.NV21ضبط کنید.