fd5fdd2507cb26d746324ae7bfd6c280580dd5e6,src/predict.py,,,#,124

Before Change

        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

After Change

    dw = 480
    dh = 480
    dataset = datasets.load(dataset_name=dataset_name)
    nc = dataset.num_classes()

    if K.backend() == "tensorflow":
        print("Tensorflow backend detected; Applying memory usage constraints")
        ss = K.tf.Session(config=K.tf.ConfigProto(gpu_options=K.tf.GPUOptions(allow_growth=True)))

Instances