towhee.models.omnivore.omnivore.omnivore_swinl_imagenet21k

towhee.models.omnivore.omnivore.omnivore_swinl_imagenet21k(pretrained: bool = True, progress: bool = True, load_heads: bool = True, **kwargs: Any) → Module[source]

Swin L patch 244 window 877 pretrained on Imagenet 1k, Imagenet 21k, Kinetics 400, SUN RGBD. By default the pretrained weights will be loaded. :param pretrained: if True loads weights from model trained on

Imagenet1k, Imagenet 21k, Kinetics 400, SUN RGBD.

Parameters:

  • progress – print progress of loading checkpoint

  • load_heads – if True, loads the 3 heads, one each for image/video/rgbd prediction. If False loads only the trunk.

Returns:

nn.Module of the omnivore model

Return type:

model