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
