b8f005eebd686c3e16b3a3603f294db0013a569e,src/predict.py,,,#,47

Before Change

        if os.path.isfile(out_file):
            continue
        pred = predict(segmenter, imfile, target_shape=(dw, dh))
        res = cv2.imwrite(out_file, pred)
    print("")

After Change

        if mode == "pillow":
            pred_final = predict_pillow(segmenter, imfile, h=dh, w=dw)
            // pred_final.save(out_file)
        elif mode == "opencv":
            pred_final = predict_opencv(segmenter, imfile, h=dh, w=dw)
            // res = cv2.imwrite(out_file, pred_final)
        elif mode == "skimage":
            pred_final = predict_skimage(segmenter, imfile, h=dh, w=dw)
            // res = io.imsave(out_file, pred_final)
        // break

    print("")

Instances