Skip to content

Instantly share code, notes, and snippets.

@jmrnilsson
jmrnilsson / simple_cb.py
Last active October 2, 2018 22:49 — forked from DavidYKay/simple_cb.py
Simple color balance algorithm using Python 2.7.8 and opencv-python=3.4.3.18. Ported from: http://www.morethantechnical.com/2015/01/14/simplest-color-balance-with-opencv-wcode/
import cv2
import math
import numpy as np
import sys
def apply_mask(matrix, mask, fill_value):
masked = np.ma.array(matrix, mask=mask, fill_value=fill_value)
return masked.filled()
def apply_threshold(matrix, low_value, high_value):