41ef0bf4ece3daf13a02b14c79eb7ddebb42bc0c,facegen/generate.py,,generate_from_yaml,#,418

Before Change

    gen = model.predict(inputs, batch_size=batch_size)

    print("Writing to "{}"...".format(output_dir))

    for i in range(0, gen.shape[0]):
        if K.image_dim_ordering() == "th":

After Change

    print("Generating images...")

    num_images = inputs["identity"].shape[0]
    count = 0

    for idx in tqdm(range(0, num_images, batch_size)):

        batch = {
            "identity":    inputs["identity"]   [idx:idx+batch_size,:],
            "emotion":     inputs["emotion"]    [idx:idx+batch_size,:],
            "orientation": inputs["orientation"][idx:idx+batch_size,:],
        }

        gen = model.predict_on_batch(batch)

        for i in range(0, gen.shape[0]):

Instances