1.github

下载想用的模型

 2.导入models

找到自己的models文件夹

例如 \torchvision\models

3.将下载好的模型导入models文件夹

4.修改文件夹中的__init__.py文件

from .bulrpool import *





from .alexnet import *
from .convnext import *
#from .resnet import *
from .pyconvresnet import *
from .resnetblurpool import *
from .blurpool import *
from .vgg import *
from .squeezenet import *
from .inception import *
from .densenet import *
from .googlenet import *
from .mobilenet import *
from .mnasnet import *
from .shufflenetv2 import *
from .efficientnet import *
from .regnet import *
from .vision_transformer import *
from . import detection
from . import feature_extraction
from . import optical_flow
from . import quantization
from . import segmentation
from . import video

6.导入该模型

model_ft = models.********(pretrained=True)

7.就完成导入了

注：如果该模型还需要其他模型辅助，根据上述步骤再进行添加