Python ResNet50RetinaNet Exemples

Exemple #1

def create_model():
	image = keras.layers.Input((512, 512, 3))
	im_info = keras.layers.Input((3,))
	gt_boxes = keras.layers.Input((None, 5))
	return ResNet50RetinaNet([image, im_info, gt_boxes])

Exemple #2

def create_model(weights='imagenet'):
    image = keras.layers.Input((None, None, 3))
    return ResNet50RetinaNet(image, num_classes=90, weights=weights)

Exemple #3

def create_model():
	image = keras.layers.Input((None, None, 3))
	gt_boxes = keras.layers.Input((None, 5))
	return ResNet50RetinaNet([image, gt_boxes], num_classes=21)