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.


nn.Module of the omnivore model

Return type: