
CvT的PyTorch实现: convolution-vision-transformers
5星
- 浏览量: 0
- 大小:None
- 文件类型:None
简介:
本文介绍了Convolution-Vision-Transformers (CvT) 的PyTorch实现方法,结合了卷积神经网络和视觉变换器的优点,适用于图像识别任务。
CvT:将卷积引入视觉变形金刚的Pytorch实现用法如下:
```python
img = torch.ones([1, 3, 224, 224])
model = CvT(224, 3, 1000)
parameters = filter(lambda p: p.requires_grad, model.parameters())
parameters = sum([np.prod(p.size()) for p in parameters]) / 1_000_000
print(Trainable Parameters: %.3fM % parameters)
out = model(img)
print(Shape of out :, out.shape) # [B, num_classes]
```
全部评论 (0)
还没有任何评论哟~


