mrtoth/imgrate
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

Refactor image loading to class imgrate

Browse Source
This commit is contained in:
Kristóf Tóth 2018-02-24 20:25:05 +01:00
parent 64561d7be4
commit 66942b9b4b
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
imgrate.py
View File

@ -16,8 +16,13 @@ def image_required(fun):
class imgrate: class imgrate:
def __init__(self, image=None): def __init__(self, imgfile=None):
self.image = image self.image = None
if imgfile: self.load_image(imgfile)
def load_image(self, imgfile):
image = cv2.imread(imgfile)
self.image = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
@image_required @image_required
def laplacian_variance(self): def laplacian_variance(self):
@ -38,7 +43,4 @@ if __name__ == '__main__':
ap.add_argument('images', type=str, nargs='+', help='') ap.add_argument('images', type=str, nargs='+', help='')
args = ap.parse_args() args = ap.parse_args()
image = cv2.imread(args.images[0]) print(imgrate(args.images[0]).laplacian_variance())
grayscale = cv2.cvtColor(image, cv2.COLOR_BGR2GRAY)
print(imgrate(grayscale).laplacian_variance())