Resim sınıflandırma, bir resmin neyi temsil ettiğini belirlemek için makine öğreniminin yaygın olarak kullanıldığı bir yöntemdir. Örneğin, belirli bir resimde hangi hayvan türünün göründüğünü öğrenmek isteyebiliriz. Bir resmin neyi temsil ettiğini tahmin etme görevine görüntü sınıflandırma adı verilir. Görüntü sınıflandırıcı, çeşitli görüntü sınıflarını tanıyacak şekilde eğitilir. Örneğin, bir model üç farklı hayvan türünü (tavşan, hamster ve köpek) temsil eden fotoğrafları tanıyacak şekilde eğitilebilir. Resim sınıflandırıcılar hakkında daha fazla bilgi için resim sınıflandırma örneğini inceleyin.
Özel görüntü sınıflandırıcılarınızı veya önceden eğitilmiş olanları mobil uygulamalarınıza dağıtmak için Görev Kitaplığı ImageClassifier API'sini kullanın.
ImageClassifier API'nin temel özellikleri
Döndürme, yeniden boyutlandırma ve renk alanı dönüştürme dahil olmak üzere giriş görüntüsü işleme.
Giriş resminin ilgi alanı.
Etiket haritası yerel ayarı.
Sonuçları filtrelemek için puan eşiği.
İlk k sınıflandırma sonucu.
Etiket izin verilenler ve red listesi.
Desteklenen görüntü sınıflandırıcı modelleri
Aşağıdaki modellerin ImageClassifier
API ile uyumlu olduğu garanti edilir.
Görüntü Sınıflandırması için TensorFlow Lite Model Maker ile oluşturulan modeller.
TensorFlow Hub'daki önceden eğitilmiş görüntü sınıflandırma modelleri.
AutoML Vision Edge Görüntü Sınıflandırma ile oluşturulan modeller.
Model uyumluluğu koşullarını karşılayan özel modeller.
Java'da çıkarım çalıştırma
Android uygulamasında ImageClassifier nasıl kullanılacağına dair bir örnek için Görüntü Sınıflandırma referans uygulamasına bakın.
1. adım: Gradle bağımlılığını ve diğer ayarları içe aktarın
.tflite model dosyasını, modelin çalıştırılacağı Android modülünün öğeler dizinine kopyalayın. Dosyanın sıkıştırılmaması gerektiğini belirtin ve TensorFlow Lite kitaplığını modülün build.gradle dosyasına ekleyin:
android {
// Other settings
// Specify tflite file should not be compressed for the app apk
aaptOptions {
noCompress "tflite"
}
}
dependencies {
// Other dependencies
// Import the Task Vision Library dependency
implementation 'org.tensorflow:tensorflow-lite-task-vision'
// Import the GPU delegate plugin Library for GPU inference
implementation 'org.tensorflow:tensorflow-lite-gpu-delegate-plugin'
}
2. adım: Modeli kullanma
// Initialization
ImageClassifierOptions options =
ImageClassifierOptions.builder()
.setBaseOptions(BaseOptions.builder().useGpu().build())
.setMaxResults(1)
.build();
ImageClassifier imageClassifier =
ImageClassifier.createFromFileAndOptions(
context, modelFile, options);
// Run inference
List<Classifications> results = imageClassifier.classify(image);
ImageClassifier yapılandırmayla ilgili daha fazla seçenek için kaynak kodu ve javadoc'a bakın.
iOS'te çıkarım çalıştırma
1. adım: Bağımlılıkları yükleyin
Görev Kitaplığı, CocoaPods kullanılarak yüklenmeyi destekler. Sisteminizde CocoaPods'un yüklü olduğundan emin olun. Talimatlar için lütfen CocoaPods yükleme kılavuzuna bakın.
Pod'ları Xcode projesine ekleme hakkında ayrıntılı bilgi için lütfen CocoaPods rehberine bakın.
Podfile'a TensorFlowLiteTaskVision kapsülünü ekleyin.
target 'MyAppWithTaskAPI' do
use_frameworks!
pod 'TensorFlowLiteTaskVision'
end
Çıkarım için kullanacağınız .tflite modelin uygulama paketinizde bulunduğundan emin olun.
2. adım: Modeli kullanma
Swift
// Imports
import TensorFlowLiteTaskVision
// Initialization
guard let modelPath = Bundle.main.path(forResource: "birds_V1",
ofType: "tflite") else { return }
let options = ImageClassifierOptions(modelPath: modelPath)
// Configure any additional options:
// options.classificationOptions.maxResults = 3
let classifier = try ImageClassifier.classifier(options: options)
// Convert the input image to MLImage.
// There are other sources for MLImage. For more details, please see:
// https://developers.google.com/ml-kit/reference/ios/mlimage/api/reference/Classes/GMLImage
guard let image = UIImage (named: "sparrow.jpg"), let mlImage = MLImage(image: image) else { return }
// Run inference
let classificationResults = try classifier.classify(mlImage: mlImage)
Objective-C
// Imports
#import <TensorFlowLiteTaskVision/TensorFlowLiteTaskVision.h>
// Initialization
NSString *modelPath = [[NSBundle mainBundle] pathForResource:@"birds_V1" ofType:@"tflite"];
TFLImageClassifierOptions *options =
[[TFLImageClassifierOptions alloc] initWithModelPath:modelPath];
// Configure any additional options:
// options.classificationOptions.maxResults = 3;
TFLImageClassifier *classifier = [TFLImageClassifier imageClassifierWithOptions:options
error:nil];
// Convert the input image to MLImage.
UIImage *image = [UIImage imageNamed:@"sparrow.jpg"];
// There are other sources for GMLImage. For more details, please see:
// https://developers.google.com/ml-kit/reference/ios/mlimage/api/reference/Classes/GMLImage
GMLImage *gmlImage = [[GMLImage alloc] initWithImage:image];
// Run inference
TFLClassificationResult *classificationResult =
[classifier classifyWithGMLImage:gmlImage error:nil];
TFLImageClassifier yapılandırmayla ilgili diğer seçenekler için kaynak koduna bakın.
Python'da çıkarım çalıştırma
1. adım: pip paketini yükleyin
pip install tflite-support
2. adım: Modeli kullanma
# Imports
from tflite_support.task import vision
from tflite_support.task import core
from tflite_support.task import processor
# Initialization
base_options = core.BaseOptions(file_name=model_path)
classification_options = processor.ClassificationOptions(max_results=2)
options = vision.ImageClassifierOptions(base_options=base_options, classification_options=classification_options)
classifier = vision.ImageClassifier.create_from_options(options)
# Alternatively, you can create an image classifier in the following manner:
# classifier = vision.ImageClassifier.create_from_file(model_path)
# Run inference
image = vision.TensorImage.create_from_file(image_path)
classification_result = classifier.classify(image)
ImageClassifier yapılandırmayla ilgili diğer seçenekler için kaynak koduna bakın.
C++'ta çıkarım çalıştırma
// Initialization
ImageClassifierOptions options;
options.mutable_base_options()->mutable_model_file()->set_file_name(model_path);
std::unique_ptr<ImageClassifier> image_classifier = ImageClassifier::CreateFromOptions(options).value();
// Create input frame_buffer from your inputs, `image_data` and `image_dimension`.
// See more information here: tensorflow_lite_support/cc/task/vision/utils/frame_buffer_common_utils.h
std::unique_ptr<FrameBuffer> frame_buffer = CreateFromRgbRawBuffer(
image_data, image_dimension);
// Run inference
const ClassificationResult result = image_classifier->Classify(*frame_buffer).value();
ImageClassifier yapılandırmayla ilgili diğer seçenekler için kaynak koduna bakın.
Örnek sonuçlar
Aşağıda, bir kuş sınıflandırıcısının sınıflandırma sonuçlarına ilişkin bir örnek verilmiştir.

Results:
Rank #0:
index : 671
score : 0.91406
class name : /m/01bwb9
display name: Passer domesticus
Rank #1:
index : 670
score : 0.00391
class name : /m/01bwbt
display name: Passer montanus
Rank #2:
index : 495
score : 0.00391
class name : /m/0bwm6m
display name: Passer italiae
ImageClassifier için basit KSA demo aracını kendi modeliniz ve test verilerinizle deneyin.
Model uyumluluğu koşulları
ImageClassifier API, zorunlu TFLite Model Metadata içeren bir TFLite modeli bekler. TensorFlow Lite Metadata Writer API'yi kullanarak görüntü sınıflandırıcılar için meta veri oluşturma örneklerini inceleyin.
Uyumlu görüntü sınıflandırıcı modelleri aşağıdaki koşulları karşılamalıdır:
Giriş görüntüsü tensörü (kTfLiteUInt8/kTfLiteFloat32)
[batch x height x width x channels]boyutunda görüntü girişi.- Toplu çıkarım desteklenmez (
batch1 olmalıdır). - Yalnızca RGB girişleri desteklenir (
channels3 olmalıdır). - Tür kTfLiteFloat32 ise giriş normalleştirme için NormalizationOptions'ın meta verilere eklenmesi gerekir.
Çıkış puanı tensörü (kTfLiteUInt8/kTfLiteFloat32)
Nsınıf ve 2 veya 4 boyutla (ör.[1 x N]veya[1 x 1 x 1 x N])- İsteğe bağlı (ancak önerilir) etiket haritaları, AssociatedFile-s olarak TENSOR_AXIS_LABELS türünde olmalıdır. Her satırda bir etiket bulunur. Örnek
etiket
dosyasına bakın.
Bu türden ilk AssociatedFile (varsa), sonuçların
labelalanını (C++'daclass_nameolarak adlandırılır) doldurmak için kullanılır.display_namealanı, oluşturma sırasında kullanılanImageClassifierOptionsöğesinindisplay_names_localealanıyla yerel ayarı eşleşen AssociatedFile (varsa) öğesinden doldurulur ("en" varsayılan olarak İngilizce'dir). Bunlardan hiçbiri yoksa sonuçların yalnızcaindexalanı doldurulur.