From 23a6c12fbef48f855791a7b73a9ff35814fb81ea Mon Sep 17 00:00:00 2001 From: Jeena Date: Wed, 8 Sep 2021 10:39:57 +0900 Subject: [PATCH] Add meta data as text overlay --- Pipfile | 3 ++ Pipfile.lock | 71 ++++++++++++++++++++++++++++++++------------- edit.py | 82 +++++++++++++++++++++++++++++++++++++++++++++++++++- 3 files changed, 135 insertions(+), 21 deletions(-) diff --git a/Pipfile b/Pipfile index 6dea604..6eb382e 100644 --- a/Pipfile +++ b/Pipfile @@ -8,6 +8,9 @@ google-auth-oauthlib = "*" psycopg2-binary = "*" google-api-python-client = "*" opencv-python = "*" +numpy = "*" +exif = "*" +geopy = "*" [dev-packages] diff --git a/Pipfile.lock b/Pipfile.lock index 73663b5..c6238a7 100644 --- a/Pipfile.lock +++ b/Pipfile.lock @@ -1,7 +1,7 @@ { "_meta": { "hash": { - "sha256": "4adb8a2006b657ec935e53c94bdbbd772d4460f29ea7e062a32b5c1685cb055b" + "sha256": "bc276a09e3e32bf9f2527093f719e6b16d5fe0db5f59e428c282089bd8e83f86" }, "pipfile-spec": 6, "requires": { @@ -39,30 +39,53 @@ "markers": "python_version >= '3'", "version": "==2.0.4" }, - "google-api-core": { + "exif": { "hashes": [ - "sha256:428805de17b48ca1af2fdb5bbfc2334e1bbcb0ea4a16506fa1337fb29c8f37c8", - "sha256:bd9eb0709f4e10dd6fddb32fd0788a190b434426c258be6e00ef40da136d768d" - ], - "markers": "python_version >= '3.6'", - "version": "==2.0.0" - }, - "google-api-python-client": { - "hashes": [ - "sha256:6e990fc4d0419c2011f75ca5c2762efbb7eb4def67bbe2f1b98a8ccb73117bf5", - "sha256:a25661ec6cf4c159f41fe9c061c2bee31b2dddaf2ad787e23617048a25b53842" + "sha256:7a93ced12bca131b04c7e92c13a16a40685b9eb83414f9ed479d38df1c686633", + "sha256:d9bae4fd7737a879402dd6e5d1675e230185e1221c3df62238faf2c8f08756d4" ], "index": "pypi", - "version": "==2.18.0" + "version": "==1.3.2" }, - "google-auth": { + "geographiclib": { "hashes": [ - "sha256:c012c8be7c442c8309ca8fa0876fef33f5fd977c467be1e1c1c2f721e8ebd73c", - "sha256:ea1af050b3e06eb73e4470f704d23007307bc0e87c13e015f6b90460f1407bd3" + "sha256:8f441c527b0b8a26cd96c965565ff0513d1e4d9952b704bf449409e5015c77b7", + "sha256:ac400d672b8954b0306bca890b088bb8ba2a757dc8133cca0b878f34b33b2740" + ], + "version": "==1.52" + }, + "geopy": { + "hashes": [ + "sha256:58b7edf526b8c32e33126570b5f4fcdfaa29d4416506064777ae8d84cd103fdd", + "sha256:8f1f949082b964385de61fcc3a667a6a9a6e242beb1ae8972449f164b2ba0e89" + ], + "index": "pypi", + "version": "==2.2.0" + }, + "google-api-core": { + "hashes": [ + "sha256:b8ad41f72a70dd709dd13a08478936172aecf5f2d34a18ab378efa6d2d6ab575", + "sha256:d6760f21b3a064a8397916b33be7383fc169d1a3c3d7fae7b47eb92bba7892b8" ], "markers": "python_version >= '3.6'", "version": "==2.0.1" }, + "google-api-python-client": { + "hashes": [ + "sha256:1f2657d1be246bc3d4bd1f0642f498653ce5fba507e6a05bbc405ed65bf10377", + "sha256:c89b345615188fbd525f52d59013156ad3bfd1023af27041f2dec3d7877ba112" + ], + "index": "pypi", + "version": "==2.19.1" + }, + "google-auth": { + "hashes": [ + "sha256:104475dc4d57bbae49017aea16fffbb763204fa2d6a70f1f3cc79962c1a383a4", + "sha256:cde472372e030e1e0bc64dac00fb53e6c095d7ab641f4281e2c995e85e205d8b" + ], + "markers": "python_version >= '3.6'", + "version": "==2.0.2" + }, "google-auth-httplib2": { "hashes": [ "sha256:31e49c36c6b5643b57e82617cb3e021e3e1d2df9da63af67252c02fa9c1f4a10", @@ -72,11 +95,11 @@ }, "google-auth-oauthlib": { "hashes": [ - "sha256:4ab58e6c3dc6ccf112f921fcced40e5426fba266768986ea502228488276eaba", - "sha256:b5a1ce7c617d247ccb2dfbba9d4bfc734b41096803d854a2c52592ae80150a67" + "sha256:3f2a6e802eebbb6fb736a370fbf3b055edcb6b52878bf2f26330b5e041316c73", + "sha256:a90a072f6993f2c327067bf65270046384cda5a8ecb20b94ea9a687f1f233a7a" ], "index": "pypi", - "version": "==0.4.5" + "version": "==0.4.6" }, "googleapis-common-protos": { "hashes": [ @@ -134,7 +157,7 @@ "sha256:f545c082eeb09ae678dd451a1b1dbf17babd8a0d7adea02897a76e639afca310", "sha256:fde50062d67d805bc96f1a9ecc0d37bfc2a8f02b937d2c50824d186aa91f2419" ], - "markers": "python_version < '3.11' and python_version >= '3.7'", + "index": "pypi", "version": "==1.21.2" }, "oauthlib": { @@ -175,6 +198,14 @@ "index": "pypi", "version": "==4.5.3.56" }, + "plum-py": { + "hashes": [ + "sha256:6d0bdf7aefa10e519d2902827dee2f96a554e55f04064c978197714b26ca1ab9", + "sha256:b02077cda2403d9dac65db38dba072a9db751129cc7b6d569837bf7a2276fd39" + ], + "markers": "python_version >= '3.6'", + "version": "==0.7.0" + }, "protobuf": { "hashes": [ "sha256:13ee7be3c2d9a5d2b42a1030976f760f28755fcf5863c55b1460fd205e6cd637", diff --git a/edit.py b/edit.py index 5ebed89..09cf6a4 100755 --- a/edit.py +++ b/edit.py @@ -2,6 +2,10 @@ import cv2 import sys +import numpy as np +import exif +from geopy.geocoders import Nominatim +from datetime import datetime def wait_with_check_closing(win_name): """ @@ -16,12 +20,41 @@ def wait_with_check_closing(win_name): if win_prop <= 0: break +def dms_coordinates_to_dd_coordinates(coordinates, coordinates_ref): + decimal_degrees = coordinates[0] + \ + coordinates[1] / 60 + \ + coordinates[2] / 3600 + if coordinates_ref == "S" or coordinates_ref == "W": + decimal_degrees = -decimal_degrees + return decimal_degrees + class Image: def __init__(self, path, w=1920, h=1080): self.w = w self.h = h self.path = path self.image = cv2.imread(self.path) + self.geolocator = Nominatim(user_agent="jeena-synology-pictures") + + def get_exif(self): + with open(self.path, 'rb') as img: + e = exif.Image(img) + if e and e.has_exif: + return e + else: + return None + + def get_place_name(self, e): + lat = str(dms_coordinates_to_dd_coordinates(e['gps_latitude'], e['gps_latitude_ref'])) + lng = str(dms_coordinates_to_dd_coordinates(e['gps_longitude'], e['gps_longitude_ref'])) + if lat and lng: + location = self.geolocator.reverse(lat + ", " + lng, language="en") + city = location.raw.get('address', {}).get('city', "") + country = location.raw.get('address', {}).get('country', "") + name = ", ".join((city, country)) + return name + else: + return "" def crop(self): oh, ow, z = self.image.shape @@ -56,8 +89,10 @@ class Image: self.image = cv2.resize(self.image, (int(w), int(h))) self.image = self.image[0:self.h, 0:self.w] bg_image = cv2.resize(self.image.copy(), (self.w, self.h)) + # make darker + bg_image = cv2.add(bg_image, np.array([-25.0])) bg_image = cv2.blur(bg_image, (200, 200)) - x_offset = int(self.image.shape[0] / 2) + x_offset = int(self.w / 2 - self.image.shape[1] / 2) y_offset = 0 bg_image[y_offset:y_offset+self.image.shape[0], x_offset:x_offset+self.image.shape[1]] = self.image self.image = bg_image @@ -69,6 +104,50 @@ class Image: if x > 0 and y > 0: self.image = self.image[int(y):int(y+self.h), int(x):int(x+self.w)] + def add_metadata(self): + e = self.get_exif() + if e: + d = datetime.strptime(e['datetime_original'], "%Y:%m:%d %H:%M:%S") + date = d.strftime("%Y-%m-%d %H:%M") + place = self.get_place_name(e) + self.writeText(date, 2) + self.writeText(place, 1) + + def writeText(self, text, line_from_bottom): + WHITE = (255, 255, 255) + BLACK = (0, 0, 0) + font = cv2.FONT_HERSHEY_SIMPLEX + font_size = 1 + font_color = BLACK + font_thickness = 4 + line_height = 1.2 + padding = 20 + textsize, baseline = cv2.getTextSize(text, font, font_size, font_thickness) + x = self.image.shape[1] - textsize[0] - padding + y = int(self.image.shape[0] - textsize[1] * line_from_bottom * line_height + baseline - padding) + print(baseline) + # outline + self.image = cv2.putText(self.image, + text, + (x,y), + font, + font_size, + font_color, + font_thickness, + cv2.LINE_AA) + # text + font_color = WHITE + font_thickness = 2 + self.image = cv2.putText(self.image, + text, + (x,y), + font, + font_size, + font_color, + font_thickness, + cv2.LINE_AA) + + def safe(self, new_path): cv2.imwrite(new_path, self.image) @@ -85,5 +164,6 @@ if __name__ == "__main__": img_path = sys.argv[1] img = Image(img_path) img.crop() + img.add_metadata() img.show() img.safe("/home/jeena/Downloads/test.jpg")