Skip to content

Latest commit

 

History

History
66 lines (47 loc) · 1.07 KB

models.rst

File metadata and controls

66 lines (47 loc) · 1.07 KB

torchgeo.models

.. module:: torchgeo.models

Change Star

.. autoclass:: ChangeStar
.. autoclass:: ChangeStarFarSeg
.. autoclass:: ChangeMixin

FarSeg

.. autoclass:: FarSeg

Fully-convolutional Network

.. autoclass:: FCN

FC Siamese Networks

.. autoclass:: FCSiamConc
.. autoclass:: FCSiamDiff

RCF Extractor

.. autoclass:: RCF

ResNet

.. autofunction:: resnet18
.. autofunction:: resnet50
.. autoclass:: ResNet18_Weights
.. autoclass:: ResNet50_Weights

Vision Transformer

.. autofunction:: vit_small_patch16_224
.. autoclass:: ViTSmall16_Weights

Utility Functions

.. autofunction:: get_model
.. autofunction:: get_model_weights
.. autofunction:: get_weight
.. autofunction:: list_models