(整理链接)常用网络的Pytorch实现
整理出来方便自用,不用每次都去到处找开源代码。(以下链接中的代码可直接拷贝使用)
CNNs
LeNet
AlexNet
VGG系列
vision/vgg.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
ResNet系列及其衍生系列
- ResNet:vision/resnet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
- ResNeSt:ResNeSt/resnet.py at 5fe47e93bd7e098d15bc278d8ab4812b82b49414 · zhanghang1989/ResNeSt (github.com)
- ResNeXt:vision/resnet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
- SENet:senet.pytorch/se_resnet.py at master · moskomule/senet.pytorch (github.com)
- SKNet:SKNet/sknet.py at master · pppLang/SKNet (github.com)
- RepVGG:RepVGG/repvgg.py at main · DingXiaoH/RepVGG (github.com)
DenseNet
vision/densenet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
GoogLeNet
vision/googlenet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
MobileNet系列
- V1:imgclsmob/mobilenet.py at 956b4ebab0bbf98de4e1548287df5197a3c7154e · osmr/imgclsmob (github.com)
- V2:vision/mobilenet.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
ShuffleNet系列
- V1:imgclsmob/shufflenet.py at c03fa67de3c9e454e9b6d35fe9cbb6b15c28fda7 · osmr/imgclsmob (github.com)
- V2:vision/shufflenetv2.py at 6db1569c89094cf23f3bc41f79275c45e9fcb3f3 · pytorch/vision (github.com)
EfficientNet系列
- V1:EfficientNet-PyTorch/model.py at 2eb7a7d264344ddf15d0a06ee99b0dca524c6a07 · lukemelas/EfficientNet-PyTorch (github.com)
- V2:automl/effnetv2_model.py at master · google/automl (github.com)
HRNet
Deformable Convolutional Network
- DCNv1:待补充
- DCNv2:待补充
FCN
vision/fcn.py at bf843c664b8ba0ff49d2921237500c77d82f2d04 · pytorch/vision (github.com)
UNet及其衍生
- U-Net:Pytorch-UNet/unet_model.py at 67bf11b4db4c5f2891bd7e8e7f58bcde8ee2d2db · milesial/Pytorch-UNet (github.com)
- U-Net++:UNetPlusPlus/generic_UNetPlusPlus.py at e145ba63862982bf1099cf2ec11d5466b434ae0b · MrGiovanni/UNetPlusPlus (github.com)
- U-Net 3+:UNet-Version/UNet_3Plus.py at master · ZJUGiveLab/UNet-Version (github.com)
- Attention U-Net:Image_Segmentation/network.py at master · LeeJunHyun/Image_Segmentation (github.com)
- R2 U-Net:Image_Segmentation/network.py at master · LeeJunHyun/Image_Segmentation (github.com)
- U^2 Net:U-2-Net/u2net.py at master · xuebinqin/U-2-Net (github.com)
SegNet
PSPNet
semseg/pspnet.py at 7192f922b99468969cfd4535e3e35a838994b115 · hszhao/semseg (github.com)
DeepLab系列
- V1:deeplab-pytorch/deeplabv1.py at master · kazuto1011/deeplab-pytorch (github.com)
- V2:deeplab-pytorch/deeplabv2.py at master · kazuto1011/deeplab-pytorch (github.com)
- V3:deeplabv3/deeplabv3.py at master · fregu856/deeplabv3 (github.com)
- V3+:deeplabv3plus-pytorch/deeplabv3plus.py at master · YudeWang/deeplabv3plus-pytorch (github.com)
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
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)