mobilenetv2 pytorch
时间: 2023-04-29 16:03:04 浏览: 96
mobilenetv2.pytorch:ImageNet上72.8%的MobileNetV2 1.0模型以及一系列预先训练的MobileNetV2模型
MobileNetV2 是一种轻量级的卷积神经网络,它在 MobileNetV1 的基础上进行了改进。在 PyTorch 中,可以使用 torchvision.models 模块中的 mobilenet_v2() 函数来加载预训练的 MobileNetV2 模型。也可以使用自己的数据训练 MobileNetV2 模型。
阅读全文