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

Add support for multiple files

Browse Source
This commit is contained in:
Kristóf Tóth 2018-02-24 20:41:57 +01:00
parent 7769250b15
commit 6f0fb68e2b
1 changed files with 6 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
imgrate.py
View File

@ -48,4 +48,9 @@ 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()
print(imgrate(args.images[0]).laplacian) ratings = {image: imgrate(image).laplacian for image in args.images}
maximum = max(ratings, key=ratings.get)
for rating in ratings.keys():
maxmark = '*' if rating == maximum and len(ratings) > 1 else ' '
print('{}imgrate("{}") = {}'.format(maxmark, rating, ratings[rating]))