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

[v2.0-alpha11] Fix JPEG sampling factor from 4:1:1 to 4:2:0 (common sub-xsampling in MP4) + add close video in video_metadata_reader

Browse Source
This commit is contained in:
Lucas MATHIEU
2024-07-03 01:41:29 +02:00
parent 7049547815
commit 8579b26955
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
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__ = "GPL-3.0-or-later" __license__ = "GPL-3.0-or-later"
__version__ = "2.0-alpha10" __version__ = "2.0-alpha11"
__maintainer__ = "Lucas MATHIEU (@campanu)" __maintainer__ = "Lucas MATHIEU (@campanu)"
__email__ = "campanu@luc-geo.fr" __email__ = "campanu@luc-geo.fr"
@@ -94,6 +94,7 @@ def video_metadata_reader(video_path: str):
video_md['width'] = video.get(cv2.CAP_PROP_FRAME_WIDTH) video_md['width'] = video.get(cv2.CAP_PROP_FRAME_WIDTH)
video_md['height'] = video.get(cv2.CAP_PROP_FRAME_HEIGHT) video_md['height'] = video.get(cv2.CAP_PROP_FRAME_HEIGHT)
video_md['frame_number'] = int(video.get(cv2.CAP_PROP_FRAME_COUNT)) video_md['frame_number'] = int(video.get(cv2.CAP_PROP_FRAME_COUNT))
video = None
return video_md return video_md
@@ -512,7 +513,7 @@ try:
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)
cv2.imwrite(image_path, frame, [cv2.IMWRITE_JPEG_QUALITY, 88, cv2.IMWRITE_JPEG_PROGRESSIVE, 1, cv2.IMWRITE_JPEG_SAMPLING_FACTOR, 0x411111]) cv2.imwrite(image_path, frame, [cv2.IMWRITE_JPEG_QUALITY, 88, cv2.IMWRITE_JPEG_PROGRESSIVE, 1, cv2.IMWRITE_JPEG_SAMPLING_FACTOR, 0x111111])
# Time tags formatting # Time tags formatting
time_shift = i * frame_sampling time_shift = i * frame_sampling