threshold.py

import cv2
import numpy as np
import os.path as path
import os

from matplotlib import pyplot as plt

def valid(file):
    _, ext = path.splitext(file)
    return ext in ['.jpg', '.png', '.jpeg']


def process(filename):
    img = cv2.imread(filename)

    blur = cv2.GaussianBlur(img, (15, 15), 0)

    mask = cv2.inRange(blur, (250, 60, 10), (255, 255, 250))
    mask = cv2.bitwise_and(cv2.medianBlur(mask, 25), mask)

    mask = np.where(mask, 0, 255)

    rgba = cv2.cvtColor(img, cv2.COLOR_RGB2RGBA)
    rgba[:, :, 3] = mask

    base, ext = path.splitext(filename)
    cv2.imwrite(path.join("masked", base + ".png"), rgba)



images = [process(file) for file in os.listdir(".") if valid(file)]