预训练深度学习模型

利用深度学习研究社区开发的模型架构。世界杯预选赛小组名单流行的模型提供了健壮的体系结构,并且不需要从头开始。

模型选择提示

有许多预训练的模型可供选择,每个模型都有权衡:

  • 大小:模型需要多少内存
    模型的最终位置将决定需要考虑多少网络规模。
    当选择部署到低内存系统时,请选择专门为此任务设计的模型。
    参见边缘部署模型
  • 准确性:在重新训练之前,模型的表现如何
    通常,对于ImageNet数据集表现良好的模型表明该模型已经学习了信息特征,并且可以在新的类似任务中表现良好。
    探索更高精度的模型
  • 预测速度:模型对新图像的预测速度
    虽然预测速度会根据硬件和批处理大小等许多因素而变化,但速度也会根据所选模型的架构和模型的大小而变化。
    比较预测速度入门的简单模型

在下面的部分中探索模型之间的权衡。

将任意模型导入MATLAB,其结构为:

>> net = networkname

即。

>> net = alexnet >> net = resnet50

如果模型尚未下载,将提供一个在MATLAB中下载模型的链接。

入门的简单模型

您可以快速迭代这些模型,并尝试不同的设置,如数据预处理步骤和训练选项。一旦你觉得哪种设置效果更好,试试更精确的网络,看看它是否能提高你的效果。

更高精度模型

探索对基于图像的工作流非常有效的模型,例如图像分类、对象检测和语义分割。

对于对象检测工作流:

DarkNet-19、DarkNet-53和ResNet-50通常被用作目标检测问题和YOLO工作流的基础。参见使用对象检测的示例Yolov2而且Yolov3

对于语义分割工作流:

使用任何预定义的网络体系结构都为训练语义分割网络提供了一个方便的起点。这些是语义分割问题中常用的层架构:

  • segnetLayers
  • unetLayers
  • unet3dLayers
  • DeepLab v3 +

请参阅使用Deeplab v3+创建语义分割网络的更多信息在这里

在MATLAB中可用的模型的完整列表

新的深度学习模型和例子

Baidu
map