整理出来方便自用,不用每次都去到处找开源代码。(以下链接中的代码可直接拷贝使用)

CNNs

LeNet

AlexNet

alexnet-pytorch/model.py at d0c1b1c52296ffcbecfbf5b17e1d1685b4ca6744 · dansuh17/alexnet-pytorch (github.com)

VGG系列

vision/vgg.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)

ResNet系列及其衍生系列

DenseNet

vision/densenet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)

GoogLeNet

vision/googlenet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)

MobileNet系列

ShuffleNet系列

EfficientNet系列

HRNet

HRNet-Image-Classification/cls_hrnet.py at 8f158719e821836e21e6cba99a3241a12a13bc41 · HRNet/HRNet-Image-Classification (github.com)

Deformable Convolutional Network

  • DCNv1:待补充
  • DCNv2:待补充

FCN

vision/fcn.py at bf843c664b8ba0ff49d2921237500c77d82f2d04 · pytorch/vision (github.com)

UNet及其衍生

SegNet

pytorch-segmentation/segnet.py at 8b8e3ee20a3aa733cb19fc158ad5d7773ed6da7f · yassouali/pytorch-segmentation (github.com)

PSPNet

semseg/pspnet.py at 7192f922b99468969cfd4535e3e35a838994b115 · hszhao/semseg (github.com)

DeepLab系列

Transformers

ViT

vit-pytorch/vit.py at main · lucidrains/vit-pytorch (github.com)

DeepViT

vit-pytorch/deepvit.py at main · lucidrains/vit-pytorch (github.com)

Cross ViT

vit-pytorch/cross_vit.py at main · lucidrains/vit-pytorch (github.com)

Swin Transformer

swin-transformer-pytorch/swin_transformer.py at master · berniwal/swin-transformer-pytorch (github.com)

Token-to-Token ViT

vit-pytorch/t2t.py at main · lucidrains/vit-pytorch (github.com)

TransUNet

TransUNet/vit_seg_modeling.py at main · Beckschen/TransUNet (github.com)

Medical Transformer

Medical-Transformer/model_codes.py at main · jeya-maria-jose/Medical-Transformer (github.com)

TransFuse

待补充

Swin-UNet

待补充

nnFormer

nnFormer/neural_network.py at main · 282857341/nnFormer (github.com)

UCTranNet

UCTransNet/UCTransNet.py at main · McGregorWwww/UCTransNet (github.com)

CrossFormer

vit-pytorch/crossformer.py at main · lucidrains/vit-pytorch (github.com)