torchversion模块介绍https://blog.csdn.net/my_kingdom/article/details/85201393
torchversion.models 包含常用的分类、检测和分割模型:
例如导入分类模型:
from torchversion.models import resnet152,vgg19_bn
导入检测模型:
from torchversion.models.detection import mask_rcnn,faster-rcnn,rpn
导入分割模型:
from torchversion.models.segmentation import fcn,deeplabV3