GRAY = 2*G-R-B

GRAY = 1.5*R-G-B

import covEX
import time
import os
from os import listdir
from os.path import isfile, join

start_time = time.time()
pathname = 'Images/'
pathsave = 'Images-converted/'

filelist = [f for f in listdir(pathname) if isfile(join(pathname, f))]

for i, val in enumerate(filelist):
    print('processing file number '+str(i))
    filename = os.path.splitext(val)[0]
    fileextension = os.path.splitext(val)[1]
    covEX.convertTofile(pathname, filename, fileextension, pathsave)

print('Shite! I'm done ;) ')
print("--- %s seconds ---" % (time.time() - start_time))

import covEX
import numpy as np
import cv2

cap = cv2.VideoCapture(0)

while(True):
    # Capture frame-by-frame
    ret, frame = cap.read()

    # Our operations on the frame come here
    plantimg = covEX.convertGreenSerious(frame)
    # Display the resulting frame
    plantimg = cv2.flip(plantimg,1)
    cv2.imshow('frame',plantimg)
    if cv2.waitKey(1) & 0xFF == ord('q'):
        break

# When everything done, release the capture
cap.release()
cv2.destroyAllWindows()

import cv2
import numpy as np
import os


def calEXG(red,green,blue):
    result = 2*green-red-blue
    return result

def calEXR(red,green,blue):
    result = 1.5*red-green-blue
    return result

def savePlants(savename,plantimg,exgimg,exgimgblurred,original,notblurred,blurred,otsu,extension):
    cv2.imwrite(savename+'/'+original+extension,plantimg)
    cv2.imwrite(savename+'/'+notblurred+extension,exgimg)
    cv2.imwrite(savename+'/'+blurred+extension,exgimgblurred)

    # ret2,th2 = cv2.threshold(exgimgblurred,0,255,cv2.THRESH_BINARY+cv2.THRESH_OTSU)
    # cv2.imwrite(savename+'/'+otsu+extension,th2)


def convertGreenSerious(rawimg):
    blue = rawimg[:,:,0]
    green = rawimg[:,:,1]
    red = rawimg[:,:,2]
    exg = 1.5*green-red-blue
    processedimg = np.where(exg > 50, exg, 0)
    processedimg = np.uint8(np.abs(processedimg))

    return processedimg

def convertGreenPlant(rawimg):
    blue = rawimg[:,:,0]
    green = rawimg[:,:,1]
    red = rawimg[:,:,2]
    exg = 2*green-red-blue
    processedimg = np.where(exg > 50, exg, 0)
    processedimg = np.uint8(np.abs(processedimg))

    return processedimg

def convertRed(rawimg):
    blue = rawimg[:,:,0]
    green = rawimg[:,:,1]
    red = rawimg[:,:,2]
    exg = 1.5*red-green-blue
    processedimg = np.where(exg > 50, exg, 0)

    return processedimg



def convertTofile(filepath,filename,extension,savepath):
    directory = filepath+filename+extension
    plantimg = cv2.imread(directory)

    exgimg = convertGreenPlant(plantimg)
    exgimgblurred = cv2.GaussianBlur(exgimg,(5,5),0)

    if not os.path.exists(savepath+filename):
        os.makedirs(savepath+filename)

    savePlants(savepath+filename,plantimg,exgimg,exgimgblurred,'original','notblurred','blurred','otsu','.png')