RegNetY

RegNetY 是一个卷积网络设计空间，具有参数简单的常规模型：深度 $d$ ，初始宽度 $w_{0} > 0$ ，和斜率 $w_{a} > 0$ ，并为每个块生成不同的块宽度 $u_{j}$ 对于每个块 $j < d$ 。 RegNet 模型类型的关键限制在于块宽度的线性参数化（设计空间仅包含具有此线性结构的模型） $u_{j} = w_{0} + w_{a}\cdot{j}$

对于 RegNetX 作者有额外的限制：我们设置 $b = 1$ （瓶颈比率）， $12 \leq d \leq 28$ ，以及 $w_{m} \geq 2$ （宽度乘数）。

对于 RegNetY，作者做了一个更改，即包含 Squeeze-and-Excitation 模块。

如何在图像上使用此模型？

加载预训练模型

>>> import timm
>>> model = timm.create_model('regnety_002', pretrained=True)
>>> model.eval()

加载和预处理图像

>>> import urllib
>>> from PIL import Image
>>> from timm.data import resolve_data_config
>>> from timm.data.transforms_factory import create_transform

>>> config = resolve_data_config({}, model=model)
>>> transform = create_transform(**config)

>>> url, filename = ("https://github.com/pytorch/hub/raw/master/images/dog.jpg", "dog.jpg")
>>> urllib.request.urlretrieve(url, filename)
>>> img = Image.open(filename).convert('RGB')
>>> tensor = transform(img).unsqueeze(0) # transform and add batch dimension

获取模型预测

>>> import torch
>>> with torch.no_grad():
...     out = model(tensor)
>>> probabilities = torch.nn.functional.softmax(out[0], dim=0)
>>> print(probabilities.shape)
>>> # prints: torch.Size([1000])

获取前 5 个预测类名

>>> # Get imagenet class mappings
>>> url, filename = ("https://raw.githubusercontent.com/pytorch/hub/master/imagenet_classes.txt", "imagenet_classes.txt")
>>> urllib.request.urlretrieve(url, filename)
>>> with open("imagenet_classes.txt", "r") as f:
...     categories = [s.strip() for s in f.readlines()]

>>> # Print top categories per image
>>> top5_prob, top5_catid = torch.topk(probabilities, 5)
>>> for i in range(top5_prob.size(0)):
...     print(categories[top5_catid[i]], top5_prob[i].item())
>>> # prints class names and probabilities like:
>>> # [('Samoyed', 0.6425196528434753), ('Pomeranian', 0.04062102362513542), ('keeshond', 0.03186424449086189), ('white wolf', 0.01739676296710968), ('Eskimo dog', 0.011717947199940681)]

将模型名称替换为您要使用的变体，例如 regnety_002。您可以在此页面顶部的模型摘要中找到 ID。

要使用此模型提取图像特征，请按照timm 特征提取示例，只需更改您要使用的模型的名称。

如何微调此模型？

您可以通过更改分类器（最后一层）来微调任何预训练模型。

>>> model = timm.create_model('regnety_002', pretrained=True, num_classes=NUM_FINETUNE_CLASSES)

要在您自己的数据集上进行微调，您必须编写一个训练循环或调整 timm 的训练脚本以使用您的数据集。

如何训练此模型？

您可以按照 timm 配方脚本从头开始训练新模型。

引用

@misc{radosavovic2020designing,
      title={Designing Network Design Spaces},
      author={Ilija Radosavovic and Raj Prateek Kosaraju and Ross Girshick and Kaiming He and Piotr Dollár},
      year={2020},
      eprint={2003.13678},
      archivePrefix={arXiv},
      primaryClass={cs.CV}
}

< > 在 GitHub 上更新