文件名称：convolution-vision-transformers:CvT的PyTorch实现
文件大小：194KB
文件格式：ZIP
更新时间：2024-04-21 09:23:22
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)