def resnet3d200(num_classes=400, pretrained='kinetics-400', **kwargs):
"""Constructs a ResNet3D-200 model."""
model = ResNet3D(Bottleneck, [3, 24, 36, 3], **kwargs)
if pretrained is not None:
settings = pretrained_settings['resnet3d200'][pretrained]
model = load_pretrained(model, num_classes, settings)
return model
def wideresnet3d50(num_classes=400, pretrained='kinetics-400',
shortcut_type='B', k=2, **kwargs):
"""Constructs a ResNet-50 model."""
model = WideResNet(WideBottleneck, [3, 4, 6, 3], num_classes=num_classes, k=k,
shortcut_type=shortcut_type, **kwargs)
if pretrained is not None:
settings = pretrained_settings['wideresnet3d50'][pretrained]
model = load_pretrained(model, num_classes, settings)
return model
def resnet3d34(num_classes=400,
pretrained='kinetics-400',
shortcut_type='A',
**kwargs):
"""Constructs a ResNet3D-34 model."""
model = ResNet3D(BasicBlock, [3, 4, 6, 3],
num_classes=num_classes,
shortcut_type=shortcut_type,
**kwargs)
if pretrained is not None:
settings = pretrained_settings['resnet3d34'][pretrained]
model = load_pretrained(model, num_classes, settings)
return model