【文件属性】:
文件名称:convolution-vision-transformers:CvT的PyTorch实现
文件大小:194KB
文件格式:ZIP
更新时间:2021-04-01 15:36:42
computer-vision transformers cnn pytorch classification
CvT:将卷积引入视觉变形金刚
CvT的Pytorch实现
用法:
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]
引
【文件预览】:
convolution-vision-transformers-master
----cvt.py(4KB)
----assets()
--------model.PNG(191KB)
----module.py(4KB)
----LICENSE(1KB)
----.gitignore(2KB)
----README.md(1014B)