Görüntü Sınıflandırma
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ınyolo 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ınyolo 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
Tahmin Et
Görüntüler üzerinde tahminleri çalıştırmak için eğitilmiş bir YOLO11n-cls modeli kullanın.
Örnek
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
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.pt
verimli 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
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.pt
vb. ü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
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
Daha fazla bilgi için Validate bölümünü ziyaret edin.