مرجع ل ultralytics/models/yolo/classify/predict.py
ملاحظه
هذا الملف متاح في https://github.com/ultralytics/ultralytics/ نقطة / الرئيسية /ultralytics/نماذج/yolo/ تصنيف / predict.py. إذا اكتشفت مشكلة ، فيرجى المساعدة في إصلاحها من خلال المساهمة في طلب 🛠️ سحب. شكرا لك 🙏!
ultralytics.models.yolo.classify.predict.ClassificationPredictor
قواعد: BasePredictor
فئة توسع فئة BasePredictor للتنبؤ استنادا إلى نموذج تصنيف.
تلاحظ
- يمكن أيضا تمرير نماذج تصنيف Torchvision إلى وسيطة "model" ، أي model = 'resnet18'.
مثل
شفرة المصدر في ultralytics/models/yolo/classify/predict.py
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
تهيئة ClassificationPredictor تعيين المهمة إلى "تصنيف".
شفرة المصدر في ultralytics/models/yolo/classify/predict.py
postprocess(preds, img, orig_imgs)
تنبؤات ما بعد العمليات لإرجاع كائنات النتائج.
شفرة المصدر في ultralytics/models/yolo/classify/predict.py
preprocess(img)
يحول صورة الإدخال إلى نوع بيانات متوافق مع النموذج.