mobilenet ssd pytorch github

MobileNet-SSD是一种轻量级的目标检测模型,其结构包含一个深度可分离卷积的MobileNet作为特征提取器,以及一个单独的检测头用于检测目标。在实际应用中,MobileNet-SSD能够实现较快的推理速度和较高的检测准确率。

如果您想在PyTorch中使用MobileNet-SSD,可以在GitHub上搜索相关的开源项目。以下是几个可能有用的GitHub链接:

github.com/qfgaohao/py…

github.com/amdegroot/s…

github.com/tonylins/py… v2的PyTorch实现,作为MobileNet-SSD的特征提取器使用。

这些开源项目提供了PyTorch实现的MobileNet-SSD,并附带了训练和测试代码、数据集准备脚本以及预训练权重文件等,方便用户快速实现MobileNet-SSD目标检测任务。

  • 3年前
  •