kopia lustrzana https://github.com/OpenDroneMap/ODM
Fix blur threshold, populate make/model, remove JPEG compression
rodzic
f6c448745d
commit
23028f3add
|
@ -231,6 +231,8 @@ class Video2Dataset:
|
||||||
piexif.ImageIFD.DateTime: elapsed_time_str,
|
piexif.ImageIFD.DateTime: elapsed_time_str,
|
||||||
piexif.ImageIFD.XResolution: (frame.shape[1], 1),
|
piexif.ImageIFD.XResolution: (frame.shape[1], 1),
|
||||||
piexif.ImageIFD.YResolution: (frame.shape[0], 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": {
|
"Exif": {
|
||||||
piexif.ExifIFD.DateTimeOriginal: elapsed_time_str,
|
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_dict["GPS"] = get_gps_location(elapsed_time, gps_coords[1], gps_coords[0], gps_coords[2])
|
||||||
|
|
||||||
exif_bytes = piexif.dump(exif_dict)
|
exif_bytes = piexif.dump(exif_dict)
|
||||||
img.save(path, exif=exif_bytes)
|
img.save(path, exif=exif_bytes, quality=95)
|
||||||
|
|
||||||
return path
|
return path
|
||||||
|
|
||||||
|
|
|
@ -135,7 +135,7 @@ class ODMLoadDatasetStage(types.ODM_Stage):
|
||||||
"input": video_files,
|
"input": video_files,
|
||||||
"output": images_dir,
|
"output": images_dir,
|
||||||
|
|
||||||
"blur_threshold": 0.2,
|
"blur_threshold": 300,
|
||||||
"distance_threshold": 10,
|
"distance_threshold": 10,
|
||||||
"black_ratio_threshold": 0.98,
|
"black_ratio_threshold": 0.98,
|
||||||
"pixel_black_threshold": 0.30,
|
"pixel_black_threshold": 0.30,
|
||||||
|
|
Ładowanie…
Reference in New Issue