参考 ultralytics/models/rtdetr/model.py
注
このファイルはhttps://github.com/ultralytics/ultralytics/blob/main/ ultralytics/models/rtdetr/model .py にあります。もし問題を発見したら、Pull Request🛠️ を投稿して修正にご協力ください。ありがとうございました!
ultralytics.models.rtdetr.model.RTDETR
ベース: Model
BaiduのRT-DETR モデル用インターフェース。このVision Transformerベースの物体検出器は、高精度のリアルタイム性能を提供します。 を提供する。効率的なハイブリッドエンコーディング、IoUを意識したクエリ選択、適応可能な推論速度をサポートする。
属性:
名称 | タイプ | 説明 |
---|---|---|
model |
str
|
訓練済みモデルへのパス。デフォルトは 'rtdetr-l.pt' です。 |
ソースコード ultralytics/models/rtdetr/model.py
task_map: dict
property
タスクと対応するUltralytics クラスを関連付けた、RT-DETR のタスクマップを返します。
リターンズ
名称 | タイプ | 説明 |
---|---|---|
dict |
dict
|
タスク名をRT-DETR モデルのUltralytics タスククラスにマッピングする辞書。 |
__init__(model='rtdetr-l.pt')
与えられた訓練済みモデルファイルでRT-DETR モデルを初期化します。.pt と .yaml 形式をサポートします。
パラメーター
名称 | タイプ | 説明 | デフォルト |
---|---|---|---|
model |
str
|
訓練済みモデルへのパス。デフォルトは 'rtdetr-l.pt' です。 |
'rtdetr-l.pt'
|
レイズ
タイプ | 説明 |
---|---|
NotImplementedError
|
モデルファイルの拡張子が'pt'、'yaml'、'yml'でない場合。 |