上传者: 42138788
|
上传时间: 2021-11-11 20:58:38
|
文件大小: 13.29MB
|
文件类型: -
MobileNetV2_pytorch_cifar
这是MobileNetv2在PyTorch中的完整实现,可以在CIFAR10,CIFAR100或您自己的数据集中进行训练。 该网络来自下面的论文
残差和线性瓶颈:用于分类,检测和细分的移动网络
在该网络中,使用了反向残差结构和深度卷积。 请参阅该论文以获取更多详细信息
用法
此项目已编译并在Python 2.7和PyTorch 0.4.0上运行。以下是一些必需的依赖项:
torch 0.4.0
torchvision 0.2.1
numpy 1.14.3
tensorboardX 1.2
使用pip首先安装它们
训练与测试
下载CIFAR10或CIFAR100数据集,或准备自己的数据集,如PyTorch中定义的数据加载器
将config.py修改为您自己的配置,例如。 更改image_size或其他
运行python main.py