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])
def create_model(weights='imagenet'): image = keras.layers.Input((None, None, 3)) return ResNet50RetinaNet(image, num_classes=90, weights=weights)
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)