towhee.models.omnivore.omnivore.omnivore_swinb_imagenet21k¶
- towhee.models.omnivore.omnivore.omnivore_swinb_imagenet21k(pretrained: bool = True, progress: bool = True, load_heads: bool = True, **kwargs: Any) Module[source]¶
Omnivore Swin B model pretrained on Imagenet 1k, Imagenet 21k, Kinetics 400, SUN RGBD. By default the pretrained weights will be loaded. :param progress: print progress of loading checkpoint :param 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
