def __init__(self, cfg, map_location=None):
super(TSMRecognizer, self).__init__()
num_segs = cfg.DATASETS.NUM_SEGS
self.backbone = build_backbone(cfg, map_location=map_location)
make_temporal_shift(self.backbone, num_segs, place='block')
self.head = build_head(cfg)
def __init__(self, cfg):
super(TSNRecognizer, self).__init__()
self.backbone = build_backbone(cfg)
self.head = build_head(cfg)
self.consensus = build_consensus(cfg)
def __init__(self, cfg):
super(I3DRecognizer, self).__init__()
self.backbone = build_backbone(cfg)
self.head = build_head(cfg)
def __init__(self, cfg, map_location=None):
super(TSNRecognizer, self).__init__()
self.backbone = build_backbone(cfg, map_location=map_location)
self.head = build_head(cfg)
self.consensus = build_consensus(cfg)