02d58e0f9f7ea0b802b887edf47309a78e2cee48,run_inference.py,,main,#,40

Before Change

        for suffix, flow_output in zip(["flow", "inv_flow"], output):
            rgb_flow = flow2rgb(args.div_flow * flow_output, max_value=args.max_flow)
            to_save = (rgb_flow * 255).astype(np.uint8).transpose(1,2,0)
            imsave(save_path/"{}{}.png".format(img1_file.namebase[:-1], suffix), to_save)


if __name__ == "__main__":
    main()

After Change

    global args, save_path
    args = parser.parse_args()

    if args.output_value == "both":
        output_string = "raw output and RGB visualization"
    elif args.output_value == "raw":
        output_string = "raw output"
    elif args.output_value == "vis":
        output_string = "RGB visualization"
    print("=> will save " + output_string)
    data_dir = Path(args.data)
    print("=> fetching img pairs in "{}"".format(args.data))
    if args.output is None:

Instances