682f3e02a81db714d17e258ce0519787e8891295,samples/latent_space_interpolation.py,,main,#Any#,75

Before Change

        img = get_image(generator, latent, args.out_depth, 1)

        // save the image:
        plt.imsave(os.path.join(args.out_dir, str(global_frame_counter) + ".png"), img)

        // increment the counter:
        global_frame_counter += 1

After Change

        latent = th.unsqueeze(latent, dim=0)

        // generate the image for this point:
        img = get_image(generator, latent, args.out_depth, 1) * 255

        if not args.video_only:
            cv2.imwrite(os.path.join(
                args.out_dir, "{:05d}.png".format(global_frame_counter)), img)

        // Make an image of unsigned 8-bit integers for OpenCV
        if(args.video_name):
            img_int = img.astype(np.uint8)
            video_out.write(img_int)

        // Increment the counter
        global_frame_counter += 1

Instances