SSDlite¶
SSDLite 模型基於 SSD: Single Shot MultiBox Detector、Searching for MobileNetV3 和 MobileNetV2: Inverted Residuals and Linear Bottlenecks 論文。
警告
檢測模組處於 Beta 階段,不保證向後相容。
模型構建器¶
以下模型構建器可用於例項化 SSD Lite 模型,可選擇是否載入預訓練權重。所有模型構建器內部均依賴於 torchvision.models.detection.ssd.SSD 基類。有關此類的更多詳細資訊,請參閱 原始碼。
|
SSDlite 模型架構,輸入尺寸為 320x320,採用 MobileNetV3 Large 作為骨幹網路,如 Searching for MobileNetV3 和 MobileNetV2: Inverted Residuals and Linear Bottlenecks 所述。 |