参考 ultralytics/models/yolo/classify/predict.py
注
このファイルはhttps://github.com/ultralytics/ultralytics/blob/main/ ultralytics/models/ yolo/classify/predict .py にあります。もし問題を発見したら、Pull Request🛠️ を投稿して修正にご協力ください。ありがとうございました!
ultralytics.models.yolo.classify.predict.ClassificationPredictor
ベース: BasePredictor
分類モデルに基づく予測のための BasePredictor クラスを拡張したクラス。
備考
- Torchvisionの分類モデルは'model'引数に渡すこともできます。
例
ソースコード ultralytics/models/yolo/classify/predict.py
__init__(cfg=DEFAULT_CFG, overrides=None, _callbacks=None)
タスクを'classify'に設定してClassificationPredictorを初期化します。
ソースコード ultralytics/models/yolo/classify/predict.py
postprocess(preds, img, orig_imgs)
予測を後処理して Results オブジェクトを返します。
ソースコード ultralytics/models/yolo/classify/predict.py
preprocess(img)
入力画像をモデルと互換性のあるデータ型に変換します。