towhee.models.omnivore.omnivoreΒΆ
Functions
Omnivore model trunk: Swin B patch (2,4,4) window (1,6,7,7) :param pretrained: if True loads weights from model trained on Imagenet 1k, Kinetics 400, SUN RGBD. :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. |
|
Omnivore swin B model trained on EPIC-KITCHENS-100 dataset :param progress: print progress of loading checkpoint |
|
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. |
|
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. :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. |
|
Swin L patch 244 window 877 trained with Kinetics 600 :param pretrained: if True loads weights from model trained on Imagenet 1k, Kinetics 600, SUN RGBD. :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. |
|
Omnivore model trunk: Swin S patch (2,4,4) window (8,7,7) :param pretrained: if True loads weights from model trained on Imagenet 1k, Kinetics 400, SUN RGBD. :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. |
|
Omnivore model trunk: Swin T patch (2,4,4) window (8,7,7) :param pretrained: if True loads weights from model trained on Imagenet 1k, Kinetics 400, SUN RGBD. :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. |
Classes
Args: |