From ef37a30e7d75c2d901c7fb488127cef8756e3eb1 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Krist=C3=B3f=20T=C3=B3th?= Date: Sat, 14 Apr 2018 18:26:56 +0200 Subject: [PATCH] Nicen coding style --- imgrate.py | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) diff --git a/imgrate.py b/imgrate.py index 512b4b4..1fc127b 100755 --- a/imgrate.py +++ b/imgrate.py @@ -1,9 +1,9 @@ #!/usr/bin/env python3 -import cv2 -import numpy as np from argparse import ArgumentParser from functools import wraps +import cv2 +import numpy as np def image_required(fun): @@ -16,11 +16,11 @@ def image_required(fun): return wrapper - class imgrate: def __init__(self, imgfile=None): self.image = None - if imgfile: self.load_image(imgfile) + if imgfile: + self.load_image(imgfile) def load_image(self, imgfile): image = cv2.imread(imgfile) @@ -47,7 +47,6 @@ class imgrate: return Th / (F.shape[0] * F.shape[1]) - if __name__ == '__main__': from os import remove from sys import exit @@ -66,8 +65,10 @@ if __name__ == '__main__': def run(args): - try: ratings = {image: getattr(imgrate(image), args.method) for image in args.images} - except AttributeError: exit('Invalid --method option!') + try: + ratings = {image: getattr(imgrate(image), args.method) for image in args.images} + except AttributeError: + exit('Invalid --method option!') maximum = max(ratings, key=ratings.get) for rating in ratings.keys():