文件名称:CeiT-pytorch:卷积增强图像变压器的实现
文件大小:668KB
文件格式:ZIP
更新时间:2024-06-06 13:59:10
classifier computer-vision pytorch transformer imagenet
CeiT:卷积增强图像变压器 这是将的非官方PyTorch实现。 训练 : python train.py -c configs/default.yaml --name "name_of_exp" 用法 : import torch from ceit import CeiT img = torch . ones ([ 1 , 3 , 224 , 224 ]) model = CeiT ( image_size = 224 , patch_size = 4 , num_classes = 100 ) out = model ( img ) print ( "Shape of out :" , out . shape ) # [B, num_classes] model = CeiT ( image_size = 224 , patch_size = 4 , nu
【文件预览】:
CeiT-pytorch-master
----ceit.py(5KB)
----utils.py(5KB)
----train.py(9KB)
----assets()
--------model1.PNG(165KB)
--------model2.PNG(232KB)
--------model3.PNG(273KB)
----module.py(6KB)
----LICENSE(1KB)
----configs()
--------default.yaml(715B)
----data_utils.py(2KB)
----.gitignore(2KB)
----README.md(1KB)