From 23028f3add7824920e23ce7f10be8d18534ea6fe Mon Sep 17 00:00:00 2001 From: Piero Toffanin Date: Tue, 24 Jan 2023 17:52:39 -0500 Subject: [PATCH] Fix blur threshold, populate make/model, remove JPEG compression --- opendm/video/video2dataset.py | 4 +++- stages/dataset.py | 2 +- 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/opendm/video/video2dataset.py b/opendm/video/video2dataset.py index 7161bd64..8f977dfa 100644 --- a/opendm/video/video2dataset.py +++ b/opendm/video/video2dataset.py @@ -231,6 +231,8 @@ class Video2Dataset: piexif.ImageIFD.DateTime: elapsed_time_str, piexif.ImageIFD.XResolution: (frame.shape[1], 1), piexif.ImageIFD.YResolution: (frame.shape[0], 1), + piexif.ImageIFD.Make: "DJI" if video_info.basename.lower().startswith("dji") else "Unknown", + piexif.ImageIFD.Model: "Unknown" }, "Exif": { piexif.ExifIFD.DateTimeOriginal: elapsed_time_str, @@ -254,7 +256,7 @@ class Video2Dataset: exif_dict["GPS"] = get_gps_location(elapsed_time, gps_coords[1], gps_coords[0], gps_coords[2]) exif_bytes = piexif.dump(exif_dict) - img.save(path, exif=exif_bytes) + img.save(path, exif=exif_bytes, quality=95) return path diff --git a/stages/dataset.py b/stages/dataset.py index cdd8b792..c7b84ab9 100644 --- a/stages/dataset.py +++ b/stages/dataset.py @@ -135,7 +135,7 @@ class ODMLoadDatasetStage(types.ODM_Stage): "input": video_files, "output": images_dir, - "blur_threshold": 0.2, + "blur_threshold": 300, "distance_threshold": 10, "black_ratio_threshold": 0.98, "pixel_black_threshold": 0.30,