6d68dca2aec44c5b1a73e6c9db5c5fbf6f731fd0,keras_retinanet/bin/train.py,,main,#Any#,454

Before Change

        anchors_dict = get_anchors_params(None)
    anchors_params = AnchorParameters(**anchors_dict)

    if args.snapshots and args.anchors:
    //save anchors configure beside models
        makedirs(args.snapshot_path)
        anchors_out = os.path.join(
            args.snapshot_path,
            "anchors.yaml"
        )
        save_anchors_params(anchors_dict,anchors_out)
 
    // create the generators
    train_generator, validation_generator = create_generators(args, backbone.preprocess_image)

    // create the model
    if args.snapshot is not None:

After Change

        print("Loading model, this may take a second...")
        model            = models.load_model(args.snapshot, backbone_name=args.backbone)
        training_model   = model
        if args.config and "anchor_parameters" in args.config:
            anchor_params = parse_anchor_parameters(args.config)
            prediction_model = retinanet_bbox(model=model, anchor_params=anchor_params)
        else:
            prediction_model = retinanet_bbox(model=model)
    else:
        weights = args.weights
        // default to imagenet if nothing else is specified
        if weights is None and args.imagenet_weights:

Instances