зеркало из https://github.com/microsoft/torchgeo.git
35 строки
703 B
Python
35 строки
703 B
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
"""TorchGeo pre-trained model repository configuration file.
|
|
|
|
* https://pytorch.org/hub/
|
|
* https://pytorch.org/docs/stable/hub.html
|
|
"""
|
|
|
|
from torchgeo.models import (
|
|
dofa_base_patch16_224,
|
|
dofa_large_patch16_224,
|
|
resnet18,
|
|
resnet50,
|
|
resnet152,
|
|
scalemae_large_patch16,
|
|
swin_v2_b,
|
|
swin_v2_t,
|
|
vit_small_patch16_224,
|
|
)
|
|
|
|
__all__ = (
|
|
'dofa_base_patch16_224',
|
|
'dofa_large_patch16_224',
|
|
'resnet18',
|
|
'resnet50',
|
|
'resnet152',
|
|
'scalemae_large_patch16',
|
|
'swin_v2_t',
|
|
'swin_v2_b',
|
|
'vit_small_patch16_224',
|
|
)
|
|
|
|
dependencies = ['timm', 'torchvision']
|