mmsegmentation的demo测试-模型加载

想学下语义分割框架,首先是跑demo,但是模型加载不熟悉,半天不知道从哪里去加载模型,并且查不到相关资料,现记一下以便后用.
在这里插入图片描述

model_urls = {
    
    
    'fcn_resnet50_coco': 'https://download.pytorch.org/models/fcn_resnet50_coco-1167a1af.pth',
    'fcn_resnet101_coco': 'https://download.pytorch.org/models/fcn_resnet101_coco-7ecb50ca.pth',
    'deeplabv3_resnet50_coco': 'https://download.pytorch.org/models/deeplabv3_resnet50_coco-cd0a2569.pth',
    'deeplabv3_resnet101_coco': 'https://download.pytorch.org/models/deeplabv3_resnet101_coco-586e9e4e.pth',
    'deeplabv3_mobilenet_v3_large_coco':
        'https://download.pytorch.org/models/deeplabv3_mobilenet_v3_large-fc3c493d.pth',
    'lraspp_mobilenet_v3_large_coco': 'https://download.pytorch.org/models/lraspp_mobilenet_v3_large-d234d4ea.pth',
}

打开网址可以直接下载模型根据指令进行测试

猜你喜欢

转载自blog.csdn.net/qq_44442727/article/details/114868873