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