lumathieu/video2geoframes.py
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 11 Wiki Activity

[v2.0-alpha3] Replacement of exif library by piexif + removal of tag xmp.author + addition of tag exif.artist and exif.software + TUI enhancement

Browse Source
This commit is contained in:
Lucas MATHIEU
2024-06-24 02:03:54 +02:00
parent 7ead19a651
commit 79c9548c4c
1 changed files with 44 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

58
video2geoframes.py
View File

@@ -8,7 +8,7 @@ Designed for contribution to street-level imagery projects like Mapillary or Pan
__author__ = "Lucas MATHIEU (@campanu)" __author__ = "Lucas MATHIEU (@campanu)"
__license__ = "AGPL-3.0-or-later" __license__ = "AGPL-3.0-or-later"
__version__ = "2.0-alpha2" __version__ = "2.0-alpha3"
__maintainer__ = "Lucas MATHIEU (@campanu)" __maintainer__ = "Lucas MATHIEU (@campanu)"
__email__ = "campanu@luc-geo.fr" __email__ = "campanu@luc-geo.fr"
@@ -18,6 +18,7 @@ import platform
from datetime import datetime, timedelta from datetime import datetime, timedelta
import cv2 import cv2
import piexif
from tomlkit import dumps, loads from tomlkit import dumps, loads
from tqdm import tqdm from tqdm import tqdm
from exif import Image, GpsAltitudeRef from exif import Image, GpsAltitudeRef
@@ -291,7 +292,9 @@ output_folder = '{}/{}'.format(output_folder, video_file_name)
existing_path(output_folder) existing_path(output_folder)
# Processes # Processes
## Frame sampling + tagging (OpenCV + exif) ## Frame sampling + tagging (OpenCV + piexif)
print('\n{}'.format(locale_toml['processing']['sampling']))
i = 0 i = 0
if timelapse == user_agree: if timelapse == user_agree:
@@ -319,25 +322,52 @@ for i in tqdm(range(cv2_tqdm_range), unit=cv2_tqdm_unit):
current_datetime = current_datetime_obj.strftime('%Y:%m:%d %H:%M:%S') current_datetime = current_datetime_obj.strftime('%Y:%m:%d %H:%M:%S')
current_subsec_time = int(int(current_datetime_obj.strftime('%f')) / 1000) current_subsec_time = int(int(current_datetime_obj.strftime('%f')) / 1000)
with open(image_path, 'rb') as image_file: # exif code
image = Image(image_file) # with open(image_path, 'rb') as image_file:
image.make = make # image = Image(image_file)
image.model = model # image.make = make
image.author = author # image.model = model
image.copyright = "{}, {}".format(author, video_start_datetime_obj.strftime('%Y')) # image.author = author
image.datetime_original = current_datetime # image.copyright = "{}, {}".format(author, video_start_datetime_obj.strftime('%Y'))
#image.offset_time_original = video_rec_timezone # image.datetime_original = current_datetime
# #image.offset_time_original = video_rec_timezone
#
# if current_subsec_time > 0 :
# image.subsec_time_original = str(current_subsec_time)
#
# with open(image_path, 'wb') as tagged_image_file:
# tagged_image_file.write(image.get_file())
if current_subsec_time > 0 : # piexif code
image.subsec_time_original = str(current_subsec_time) image_exif = piexif.load(image_path)
with open(image_path, 'wb') as tagged_image_file: image_tags = {
tagged_image_file.write(image.get_file()) piexif.ImageIFD.Make: make,
piexif.ImageIFD.Model: model,
piexif.ImageIFD.Artist: author,
piexif.ImageIFD.Copyright: "{}, {}".format(author, video_start_datetime_obj.strftime('%Y')),
piexif.ImageIFD.Software : 'video2geoframes.py'
}
exif_tags = {
piexif.ExifIFD.DateTimeOriginal: current_datetime,
piexif.ExifIFD.OffsetTimeOriginal: video_rec_timezone
}
if current_subsec_time > 0:
exif_tags[piexif.ExifIFD.SubSecTime] = str(current_subsec_time)
image_exif['0th'] = image_tags
image_exif['Exif'] = exif_tags
image_exif_bytes = piexif.dump(image_exif)
piexif.insert(image_exif_bytes, image_path)
i += 1 i += 1
# Geo-tagging (ExifTool) # Geo-tagging (ExifTool)
print('\n{}'.format(locale_toml['processing']['geotagging'])) print('\n{}'.format(locale_toml['processing']['geotagging']))
geotagging_cmd = '{} -P -geotag "{}" "-geotime<SubSecDateTimeOriginal" -overwrite_original "{}/{}_f*.jpg"'\ geotagging_cmd = '{} -P -geotag "{}" "-geotime<SubSecDateTimeOriginal" -overwrite_original "{}/{}_f*.jpg"'\
.format(exiftool_path, gps_track_path, output_folder, video_file_name.split('.')[0]) .format(exiftool_path, gps_track_path, output_folder, video_file_name.split('.')[0])
geotagging = os.system(geotagging_cmd) geotagging = os.system(geotagging_cmd)