bug in predict.py script

[Laetij]

Predict function doesn't work in command line but well in the web app interface

You should replace :

Open and read the image to predict.

with tf.gfile.Open(path, 'rb') as f:
    try:
        image = Image.open(f).convert('RGB')
    except (tf.errors.OutOfRangeError, OSError) as e:
        click.echo()
        click.echo('Error while processing {}: {}'.format(path, e))
        return

by :

Open and read the image to predict.

with tf.gfile.Open(path, 'rb') as f:
    try:
        image = Image.open(f).convert('RGB')
        img = np.asarray(image)
        img = cv2.cvtColor(img, cv2.COLOR_BGR2GRAY)
        b = cv2.distanceTransform(img, distanceType=cv2.DIST_L2, maskSize=5)
        g = cv2.distanceTransform(img, distanceType=cv2.DIST_L1, maskSize=5)
        r = cv2.distanceTransform(img, distanceType=cv2.DIST_C, maskSize=5)

        # merge the transformed channels back to an image
        transformed_image = cv2.merge((b, g, r))
    except (tf.errors.OutOfRangeError, OSError) as e:
        click.echo()
        click.echo('Error while processing {}: {}'.format(path, e))
        return

to be able to use tabulo predict in command line