cloud_detection.py 文件源码

def detect(self, image, mask = None):
        floatimage = np.float32(image)

        fb,fg,fr = cv2.split(floatimage)

        nonzero = (fr + fb) != 0
        difference = np.zeros(fr.shape, np.float32)
        difference[nonzero] = (fb[nonzero] - fr[nonzero]) / (fb[nonzero] + fr[nonzero])
        _, result = cv2.threshold(difference, Configuration.nbr_threshold, 1, cv2.THRESH_BINARY_INV)
        return np.uint8(result)