MobileNetV3( inverted_residual_setting: List[InvertedResidualConfig], last_channel: int, num_classes: int = 1000, block: Optional[Callable[..., nn.Module]] = None, norm_layer: Optional[Callable[..., nn.Module]] = None)
时间: 2023-06-14 17:06:30 浏览: 179
mobilenetv3-segmentation:MobileNetV3用于语义细分
MobileNetV3是一种轻量级卷积神经网络模型,用于图像分类、目标检测等任务。它由多个倒残差块组成,每个倒残差块包含多个轻量级卷积层和激活函数,可以有效地减少模型的计算量和参数量。
参数inverted_residual_setting是一个包含多个InvertedResidualConfig的列表,每个InvertedResidualConfig包含了该倒残差块的一些参数,比如输入通道数、输出通道数、中间层的扩展因子等。
参数last_channel指定了模型最后输出的通道数。
参数num_classes指定了模型最后分类的类别数。
参数block指定了使用的卷积块类型,默认为None。
参数norm_layer指定了使用的归一化层类型,默认为None。
阅读全文