文件名称:MutualNet:[ECCV'20口头] MutualNet
文件大小:967KB
文件格式:ZIP
更新时间:2024-04-09 10:26:21
Python
MutualNet:通过相互学习从网络宽度和分辨率(ECCV'20口头)获得的自适应ConvNet 这项工作提出了一种训练网络的方法,该网络在运行时可以在动态资源约束(例如FLOP)下执行。所提出的用于输入分辨率和网络宽度的相互学习方案极大地提高了在各种任务(例如图像分类,对象检测和实例分割)上的准确性-效率折衷。该方法还有望用作即插即用策略来增强单个网络。它的性能大大优于一组流行的数据扩充和正则化方法。总体框架如下所示。 我们为不同的子网提供不同的缩放图像。如下所示,这使每个子网都可以从网络规模和输入规模中学习多规模表示。 安装 PyTorch 1.0.1,Torchvision 0.2.2,Numpy,pyyaml 5.1。 按照PyTorch准备ImageNet数据集。 跑 训练 要训练MobileNet v1,请运行以下代码 python train.py app:apps/m
【文件预览】:
MutualNet-master
----.gitignore(17B)
----models()
--------slimmable_ops.py(6KB)
--------mobilenet_v1.py(3KB)
--------mobilenet_v2.py(4KB)
--------wideresnet.py(4KB)
--------resnet.py(4KB)
----LICENSE(1KB)
----utils()
--------transforms.py(3KB)
--------datasets.py(3KB)
--------setlogger.py(639B)
--------model_profiling.py(6KB)
--------meters.py(2KB)
--------config.py(5KB)
----README.md(7KB)
----imgs()
--------mutuallearning.PNG(436KB)
--------framework.PNG(405KB)
--------result1.JPG(99KB)
--------result2.JPG(36KB)
----apps()
--------resnet50.yml(660B)
--------wideresnet.yml(550B)
--------mobilenet_v2.yml(663B)
--------mobilenet_v1.yml(734B)
----ComputePostBN.py(2KB)
----train.py(10KB)