def resnet34_ssdtc(num_classes, img_size, phase): extras = add_extras(cfg['extras'], 512) model = ResNet_SSDTC(BasicBlock, [3, 4, 6, 3], extras, num_classes, img_size, phase) return model
def xception_H(num_classes, img_size, phase): extras = add_extras(cfg['extras'], 1024) model = Xception_H(extras, num_classes, img_size, phase) return model
def nasnetmobile(num_classes, img_size, phase): extras = add_extras(cfg['extras'], in_channels=1056) model = NASNetAMobile(extras, num_classes, img_size, phase) return model
def vgg16(num_classes, img_size, phase): base = make_layers(cfg['base']) extras = add_extras(cfg['extras'], in_channels=1024) head = multibox(base, extras, cfg['mbox'], num_classes) return VGG_SSD(base, extras, head, num_classes, img_size, phase)
def resnet152(num_classes, img_size, phase): extras = add_extras(cfg['extras'], in_channels=2048) model = ResNet_SSD(Bottleneck, [3, 8, 36, 3], extras, num_classes, img_size, phase) return model
def resnet34(num_classes, img_size, phase): extras = add_extras(cfg['extras'], in_channels=512) model = ResNet_SSD(BasicBlock, [3, 4, 6, 3], extras, num_classes, img_size, phase) return model
def resnet34_features(): extras = add_extras(cfg['extras'], 512) model = ResNet_features(BasicBlock, [3, 4, 6, 3], extras) return model