4 Commits

3 changed files with 41 additions and 27 deletions

View File

@@ -1,7 +1,7 @@
# Localization file for video2geoframes.py script # Localization file for video2geoframes.py script
# English (US / World) # English (US / World)
# #
# Last edition : 2024-06-22 # Last edition : 2024-06-23
[ui] [ui]
@@ -28,10 +28,10 @@ timelapse = "Timelapse video ({}/{}) ? "
timelapse_fps = "Timelapse framerate (frame/s) [{}-{}] : " timelapse_fps = "Timelapse framerate (frame/s) [{}-{}] : "
timelapse_fps_err = "Error... please enter a decimal between {} et {}." timelapse_fps_err = "Error... please enter a decimal between {} et {}."
frame_samp = "Enter the frame sampling in seconds [{}-{}] : " frame_samp = "Enter frame sampling in seconds [{}-{}] : "
frame_samp_err = "Error... please enter a decimal between {} and {}." frame_samp_err = "Error... please enter a decimal between {} and {}."
frame_height = "Enter frame height in pixels (ratio unchanged) [{}-{}] : " frame_height = "Enter output frame height in pixels (ratio unchanged) [{}-{}] : "
frame_height_err = "Error... please enter an integer between {} and {}." frame_height_err = "Error... please enter an integer between {} and {}."
video_start_datetime = "Enter video start datetime following ISO format (exemple : 2023-09-18T22:00:02.000) : " video_start_datetime = "Enter video start datetime following ISO format (exemple : 2023-09-18T22:00:02.000) : "

View File

@@ -1,7 +1,7 @@
# Localization file for video2geoframes.py script # Localization file for video2geoframes.py script
# French (France) # French (France)
# #
# Last edition : 2024-06-22 # Last edition : 2024-06-23
[ui] [ui]
@@ -28,10 +28,10 @@ timelapse = "Vidéo timelapse ({}/{}) ? "
timelapse_fps = "Débit d'image du timelapse (image/s) [{}-{}] : " timelapse_fps = "Débit d'image du timelapse (image/s) [{}-{}] : "
timelapse_fps_err = "Erreur... entrez un entier entre {} et {}." timelapse_fps_err = "Erreur... entrez un entier entre {} et {}."
frame_samp = "Entrez l'espacement temporel des images en secondes [{}-{}] : " frame_samp = "Entrez l'espacement temporel en secondes entre les images [{}-{}] : "
frame_samp_err = "'Erreur... veuillez entrer un nombre décimal entre {} et {}." frame_samp_err = "'Erreur... veuillez entrer un nombre décimal entre {} et {}."
frame_height = "Entrez la hauteur des images en pixels (ratio inchangé) [{}-{}] : " frame_height = "Entrez la hauteur en pixels des images en sortie (ratio inchangé) [{}-{}] : "
frame_height_err = "Erreur... veuillez entrer un nombre entier entre {} et {}." frame_height_err = "Erreur... veuillez entrer un nombre entier entre {} et {}."
video_start_datetime = "Entrez l'horodatage du début de la vidéo au format ISO (exemple : 2023-09-18T22:00:02.000) : " video_start_datetime = "Entrez l'horodatage du début de la vidéo au format ISO (exemple : 2023-09-18T22:00:02.000) : "

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-alpha5" __version__ = "2.0-alpha7"
__maintainer__ = "Lucas MATHIEU (@campanu)" __maintainer__ = "Lucas MATHIEU (@campanu)"
__email__ = "campanu@luc-geo.fr" __email__ = "campanu@luc-geo.fr"
@@ -50,7 +50,7 @@ def byte_multiple(size):
# Start # Start
print('# video2geoframes.py') print('# video2geoframes.py ({})'.format(__version__))
# Configuration settings # Configuration settings
base_path = unix_path(os.path.dirname(__file__)) base_path = unix_path(os.path.dirname(__file__))
@@ -60,9 +60,13 @@ ini_file_err = False
## Default values ## Default values
locale = 'en_us' locale = 'en_us'
min_frame_samp = 0.5 min_frame_samp = 0.5
max_frame_samp = float(60) max_frame_samp = 60.0
min_timelapse_fps = 1 min_timelapse_fps = 1
max_timelapse_fps = 15 max_timelapse_fps = 15
min_frame_height = 480
max_frame_height = 9000
min_time_offset = -10.0
max_time_offset = 10.0
## Platform-dependent commands ## Platform-dependent commands
if platform.system() == 'Windows': if platform.system() == 'Windows':
@@ -154,6 +158,13 @@ else:
print('{}\n'.format(locale_toml['ui']['paths']['path_err'])) print('{}\n'.format(locale_toml['ui']['paths']['path_err']))
True True
### Video metadatas extraction
video = cv2.VideoCapture(video_path)
video_fps = video.get(cv2.CAP_PROP_FPS)
video_width = video.get(cv2.CAP_PROP_FRAME_WIDTH)
video_height = video.get(cv2.CAP_PROP_FRAME_HEIGHT)
video_total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
### GPS track file ### GPS track file
while True: while True:
try: try:
@@ -195,7 +206,6 @@ else:
else: else:
### Frame sampling parameter ### Frame sampling parameter
while True: while True:
try: try:
frame_sampling = float(input(locale_toml['ui']['parameters']['frame_samp'].format(min_frame_samp, frame_sampling = float(input(locale_toml['ui']['parameters']['frame_samp'].format(min_frame_samp,
@@ -211,8 +221,8 @@ else:
True True
## Frame height parameter ## Frame height parameter
min_frame_height = 480 if video_height <= max_frame_height:
max_frame_height = 6000 max_frame_height = int(round(video_height, 0))
while True: while True:
try: try:
@@ -221,6 +231,8 @@ else:
if max_frame_height >= frame_height >= min_frame_height: if max_frame_height >= frame_height >= min_frame_height:
break break
elif frame_height == 0:
break
else: else:
print(locale_toml['ui']['parameters']['frame_height_err'].format(min_frame_height, max_frame_height)) print(locale_toml['ui']['parameters']['frame_height_err'].format(min_frame_height, max_frame_height))
True True
@@ -242,12 +254,10 @@ else:
video_rec_timezone = input(locale_toml['ui']['parameters']['rec_timezone']) video_rec_timezone = input(locale_toml['ui']['parameters']['rec_timezone'])
### Time offset parameter ### Time offset parameter
min_time_offset = -10.0
max_time_offset = 10.0
while True: while True:
try: try:
time_offset = float(input(locale_toml['ui']['parameters']['time_offset'].format(min_time_offset, max_time_offset))) time_offset = float(input(locale_toml['ui']['parameters']['time_offset'].format(min_time_offset,
max_time_offset)))
if max_time_offset >= frame_sampling >= min_time_offset: if max_time_offset >= frame_sampling >= min_time_offset:
break break
@@ -258,7 +268,9 @@ else:
print(locale_toml['ui']['parameters']['time_offset_err'].format(min_time_offset, max_time_offset)) print(locale_toml['ui']['parameters']['time_offset_err'].format(min_time_offset, max_time_offset))
True True
### User-defined metadata ## User-defined metadata
print('\n{}'.format(locale_toml['ui']['info']['tags_title']))
make = input(locale_toml['ui']['metadatas']['make']) make = input(locale_toml['ui']['metadatas']['make'])
model = input(locale_toml['ui']['metadatas']['model']) model = input(locale_toml['ui']['metadatas']['model'])
author = input(locale_toml['ui']['metadatas']['author']) author = input(locale_toml['ui']['metadatas']['author'])
@@ -266,25 +278,19 @@ else:
# Video metadatas formatting # Video metadatas formatting
print('\n{}'.format(locale_toml['processing']['reading_metadatas'])) print('\n{}'.format(locale_toml['processing']['reading_metadatas']))
video = cv2.VideoCapture(video_path)
video_fps = video.get(cv2.CAP_PROP_FPS)
video_width = video.get(cv2.CAP_PROP_FRAME_WIDTH)
video_height = video.get(cv2.CAP_PROP_FRAME_HEIGHT)
video_total_frames = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
video_file_name = os.path.basename(video_path) video_file_name = os.path.basename(video_path)
video_file_size = byte_multiple(os.stat(video_path).st_size) video_file_size = byte_multiple(os.stat(video_path).st_size)
video_duration = video_total_frames / video_fps video_duration = video_total_frames / video_fps
video_start_datetime_obj = video_start_datetime_obj + timedelta(seconds=time_offset) video_start_datetime_obj = video_start_datetime_obj + timedelta(seconds=time_offset)
video_start_datetime = video_start_datetime_obj.strftime('%Y-%m-%d %H:%M:%S') video_start_datetime = video_start_datetime_obj.strftime('%Y-%m-%d %H:%M:%S')
video_start_subsectime = video_start_datetime_obj.strftime('%f') video_start_subsectime = int(video_start_datetime_obj.strftime('%f') / 1000)
# Metadata recap # Metadata recap
print('\n{}'.format(locale_toml['ui']['info']['metadatas'].format(video_file_name, print('\n{}'.format(locale_toml['ui']['info']['metadatas'].format(video_file_name,
round(video_file_size[0], 3), video_file_size[1], round(video_file_size[0], 3), video_file_size[1],
video_duration, video_start_datetime, video_duration, video_start_datetime,
int(int(video_start_subsectime) / 1000), '{:03d}'.format(video_start_subsectime),
video_rec_timezone))) video_rec_timezone)))
# Output folder creation # Output folder creation
@@ -310,6 +316,14 @@ for i in tqdm(range(cv2_tqdm_range), unit=cv2_tqdm_unit):
video.set(cv2.CAP_PROP_POS_MSEC, t) video.set(cv2.CAP_PROP_POS_MSEC, t)
ret, frame = video.read() ret, frame = video.read()
### Image resizing
if frame_height != 0:
resize_factor = video_height / frame_height
image_height = frame_height
image_width = int(round(video_height * resize_factor), 0)
frame = cv2.resize(frame, (image_width, image_height), interpolation=cv2.INTER_LANCZOS4)
frame_name = '{:05d}'.format(i) frame_name = '{:05d}'.format(i)
image_name = "{}_f{}.jpg".format(video_file_name.split('.')[0], frame_name) image_name = "{}_f{}.jpg".format(video_file_name.split('.')[0], frame_name)
image_path = "{}/{}".format(output_folder, image_name) image_path = "{}/{}".format(output_folder, image_name)
@@ -332,7 +346,7 @@ for i in tqdm(range(cv2_tqdm_range), unit=cv2_tqdm_unit):
# image.datetime_original = current_datetime # image.datetime_original = current_datetime
# #image.offset_time_original = video_rec_timezone # #image.offset_time_original = video_rec_timezone
# #
# if current_subsec_time > 0 : # if current_subsec_time > 0:
# image.subsec_time_original = str(current_subsec_time) # image.subsec_time_original = str(current_subsec_time)
# #
# with open(image_path, 'wb') as tagged_image_file: # with open(image_path, 'wb') as tagged_image_file:
@@ -346,7 +360,7 @@ for i in tqdm(range(cv2_tqdm_range), unit=cv2_tqdm_unit):
piexif.ImageIFD.Model: model, piexif.ImageIFD.Model: model,
piexif.ImageIFD.Artist: author, piexif.ImageIFD.Artist: author,
piexif.ImageIFD.Copyright: "{}, {}".format(author, video_start_datetime_obj.strftime('%Y')), piexif.ImageIFD.Copyright: "{}, {}".format(author, video_start_datetime_obj.strftime('%Y')),
piexif.ImageIFD.Software : 'video2geoframes.py (v{})'.format(__version__) piexif.ImageIFD.Software: 'video2geoframes.py (v{})'.format(__version__)
} }
exif_tags = { exif_tags = {