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 中所述。 |