İçeriğe geç

Görüntü Sınıflandırma

Görüntü sınıflandırma örnekleri

Görüntü sınıflandırma, üç görevden en basit olanıdır ve tüm bir görüntünün önceden tanımlanmış bir dizi sınıftan birine sınıflandırılmasını içerir.

Bir görüntü sınıflandırıcının çıktısı tek bir sınıf etiketi ve bir güven puanıdır. Görüntü sınıflandırma, bir görüntünün yalnızca hangi sınıfa ait olduğunu bilmeniz gerektiğinde ve bu sınıftaki nesnelerin nerede bulunduğunu veya tam şekillerinin ne olduğunu bilmeniz gerekmediğinde kullanışlıdır.



İzle: Ultralytics YOLO Görevlerini keşfedin: Ultralytics HUB kullanarak Görüntü Sınıflandırma

İpucu

YOLO11 Sınıflandırma modelleri -cls son ek, yani yolo11n-cls.pt ve önceden eğitilmiş ImageNet.

Modeller

YOLO11 ön eğitimli Classify modelleri burada gösterilmektedir. Detect, Segment ve Pose modelleri COCO veri kümesi üzerinde, Classify modelleri ise ImageNet veri kümesi üzerinde ön eğitime tabi tutulmuştur.

Modeller ilk kullanımda en son Ultralytics sürümünden otomatik olarak indirilir.

Model boyut
(piksel)
acc
top1
acc
top5
Hız
CPU ONNX
(ms)
Hız
T4TensorRT10
(ms)
params
(M)
FLOPs
(B) 640'da
YOLO11n-cls 224 70.0 89.4 5.0 ± 0.3 1.1 ± 0.0 1.6 3.3
YOLO11s-cls 224 75.4 92.7 7.9 ± 0.2 1.3 ± 0.0 5.5 12.1
YOLO11m-cls 224 77.3 93.9 17.2 ± 0.4 2.0 ± 0.0 10.4 39.3
YOLO11l-cls 224 78.3 94.3 23.2 ± 0.3 2.8 ± 0.0 12.9 49.4
YOLO11x-cls 224 79.5 94.9 41.4 ± 0.9 3.8 ± 0.0 28.4 110.4
  • acc değerleri model doğruluklarıdır. ImageNet veri kümesi doğrulama seti.
    Tarafından çoğaltın yolo val classify data=path/to/ImageNet device=0
  • Hız kullanarak ImageNet val görüntüleri üzerinde ortalama Amazon EC2 P4d örnek.
    Tarafından çoğaltın yolo val classify data=path/to/ImageNet batch=1 device=0|cpu

Tren

YOLO11n-cls'yi 64 görüntü boyutunda 100 epok için MNIST160 veri kümesi üzerinde eğitin. Kullanılabilir bağımsız değişkenlerin tam listesi için Yapılandırma sayfasına bakın.

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.yaml")  # build a new model from YAML
model = YOLO("yolo11n-cls.pt")  # load a pretrained model (recommended for training)
model = YOLO("yolo11n-cls.yaml").load("yolo11n-cls.pt")  # build from YAML and transfer weights

# Train the model
results = model.train(data="mnist160", epochs=100, imgsz=64)
# Build a new model from YAML and start training from scratch
yolo classify train data=mnist160 model=yolo11n-cls.yaml epochs=100 imgsz=64

# Start training from a pretrained *.pt model
yolo classify train data=mnist160 model=yolo11n-cls.pt epochs=100 imgsz=64

# Build a new model from YAML, transfer pretrained weights to it and start training
yolo classify train data=mnist160 model=yolo11n-cls.yaml pretrained=yolo11n-cls.pt epochs=100 imgsz=64

Veri kümesi formatı

YOLO sınıflandırma veri kümesi formatı ayrıntılı olarak Veri Kümesi Kılavuzunda bulunabilir.

Val

Eğitilmiş YOLO11n-cls modelini doğrulayın doğruluk MNIST160 veri kümesi üzerinde. Hiçbir argümana gerek yoktur çünkü model eğitimini sürdürür data ve model öznitelikleri olarak argümanlar.

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom model

# Validate the model
metrics = model.val()  # no arguments needed, dataset and settings remembered
metrics.top1  # top1 accuracy
metrics.top5  # top5 accuracy
yolo classify val model=yolo11n-cls.pt  # val official model
yolo classify val model=path/to/best.pt  # val custom model

Tahmin Et

Görüntüler üzerinde tahminleri çalıştırmak için eğitilmiş bir YOLO11n-cls modeli kullanın.

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom model

# Predict with the model
results = model("https://ultralytics.com/images/bus.jpg")  # predict on an image
yolo classify predict model=yolo11n-cls.pt source='https://ultralytics.com/images/bus.jpg'  # predict with official model
yolo classify predict model=path/to/best.pt source='https://ultralytics.com/images/bus.jpg'  # predict with custom model

Tamamı görün predict modunun ayrıntıları Tahmin Et Sayfa.

İhracat

Bir YOLO11n-cls modelini ONNX, CoreML vb. gibi farklı bir formata aktarın.

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load an official model
model = YOLO("path/to/best.pt")  # load a custom trained model

# Export the model
model.export(format="onnx")
yolo export model=yolo11n-cls.pt format=onnx  # export official model
yolo export model=path/to/best.pt format=onnx  # export custom trained model

Kullanılabilir YOLO11-cls dışa aktarma biçimleri aşağıdaki tabloda yer almaktadır. kullanarak herhangi bir formata dışa aktarabilirsiniz. format argümanı, yani format='onnx' veya format='engine'. Doğrudan dışa aktarılan modeller üzerinde tahmin veya doğrulama yapabilirsiniz, örn. yolo predict model=yolo11n-cls.onnx. Dışa aktarma tamamlandıktan sonra modeliniz için kullanım örnekleri gösterilir.

Biçim format Tartışma Model Metadata Argümanlar
PyTorch - yolo11n-cls.pt -
TorchScript torchscript yolo11n-cls.torchscript imgsz, optimize, batch
ONNX onnx yolo11n-cls.onnx imgsz, half, dynamic, simplify, opset, batch
OpenVINO openvino yolo11n-cls_openvino_model/ imgsz, half, int8, batch
TensorRT engine yolo11n-cls.engine imgsz, half, dynamic, simplify, workspace, int8, batch
CoreML coreml yolo11n-cls.mlpackage imgsz, half, int8, nms, batch
TF SavedModel saved_model yolo11n-cls_saved_model/ imgsz, keras, int8, batch
TF GraphDef pb yolo11n-cls.pb imgsz, batch
TF Lite tflite yolo11n-cls.tflite imgsz, half, int8, batch
TF Kenar TPU edgetpu yolo11n-cls_edgetpu.tflite imgsz
TF.js tfjs yolo11n-cls_web_model/ imgsz, half, int8, batch
PaddlePaddle paddle yolo11n-cls_paddle_model/ imgsz, batch
MNN mnn yolo11n-cls.mnn imgsz, batch, int8, half
NCNN ncnn yolo11n-cls_ncnn_model/ imgsz, half, batch
IMX500 imx yolo11n-cls_imx_model/ imgsz, int8

Tamamı görün export detaylar İhracat Sayfa.

SSS

Görüntü sınıflandırmada YOLO11 'un amacı nedir?

YOLO11 gibi modeller yolo11n-cls.ptverimli görüntü sınıflandırması için tasarlanmıştır. Bir güven puanı ile birlikte tüm görüntüye tek bir sınıf etiketi atarlar. Bu, özellikle görüntü içindeki nesnelerin konumunu veya şeklini belirlemek yerine, bir görüntünün belirli sınıfını bilmenin yeterli olduğu uygulamalar için kullanışlıdır.

Görüntü sınıflandırması için bir YOLO11 modelini nasıl eğitebilirim?

Bir YOLO11 modelini eğitmek için Python veya CLI komutlarından birini kullanabilirsiniz. Örneğin, bir modeli eğitmek için yolo11n-cls modelini 64 görüntü boyutunda 100 epok için MNIST160 veri kümesi üzerinde test etmiştir:

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load a pretrained model (recommended for training)

# Train the model
results = model.train(data="mnist160", epochs=100, imgsz=64)
yolo classify train data=mnist160 model=yolo11n-cls.pt epochs=100 imgsz=64

Daha fazla yapılandırma seçeneği için Yapılandırma sayfasını ziyaret edin.

Önceden eğitilmiş YOLO11 sınıflandırma modellerini nerede bulabilirim?

Önceden eğitilmiş YOLO11 sınıflandırma modelleri şurada bulunabilir Modeller Bölüm. Gibi modeller yolo11n-cls.pt, yolo11s-cls.pt, yolo11m-cls.ptvb. üzerinde önceden eğitilmiştir. ImageNet veri kümesidir ve kolayca indirilebilir ve çeşitli görüntü sınıflandırma görevleri için kullanılabilir.

Eğitilmiş bir YOLO11 modelini farklı formatlara nasıl aktarabilirim?

Eğitilmiş bir YOLO11 modelini Python veya CLI komutlarını kullanarak çeşitli formatlara aktarabilirsiniz. Örneğin, bir modeli ONNX biçiminde dışa aktarmak için:

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load the trained model

# Export the model to ONNX
model.export(format="onnx")
yolo export model=yolo11n-cls.pt format=onnx  # export the trained model to ONNX format

Ayrıntılı dışa aktarma seçenekleri için Dışa Aktar sayfasına bakın.

Eğitilmiş bir YOLO11 sınıflandırma modelini nasıl doğrulayabilirim?

Eğitilmiş bir modelin doğruluğunu MNIST160 gibi bir veri kümesi üzerinde doğrulamak için aşağıdaki Python veya CLI komutlarını kullanabilirsiniz:

Örnek

from ultralytics import YOLO

# Load a model
model = YOLO("yolo11n-cls.pt")  # load the trained model

# Validate the model
metrics = model.val()  # no arguments needed, uses the dataset and settings from training
metrics.top1  # top1 accuracy
metrics.top5  # top5 accuracy
yolo classify val model=yolo11n-cls.pt  # validate the trained model

Daha fazla bilgi için Validate bölümünü ziyaret edin.

📅1 yıl önce oluşturuldu ✏️ 1 ay önce güncellendi

Yorumlar