python 2 大疆照片rtk 改变分辨率重新拷贝GPS信息
给参数
default_input_dir = r"D:\0ubuntu20\1slam\数据\2RTK\City4_outskirts2\map_cloudy_0303_15pm_135m\images_source"
default_output_dir = r"D:\0ubuntu20\1slam\数据\2RTK\City4_outskirts2\map_cloudy_0303_15pm_135m\images"
default_resolution = (1805, 1203)
default_start_id = 0
"""
楂樿川閲忓浘鐗囩缉鏀捐剼鏈紙鏀寔鑷畾涔夌紪鍙烽噸鍛藉悕锛?鍔熻兘锛?- OpenCV 楂樿川閲忕缉鏀?- 瀹屾暣淇濈暀 EXIF
- 澶氳繘绋嬪姞閫?- 杈撳嚭鏂囦欢鍚嶏細PREFIX_XXXXX.jpg
"""
# pip install "numpy<2" opencv-python-headless pillow piexif tqdm
import cv2
import piexif
import numpy as np
from PIL import Image
import argparse
from pathlib import Path
import multiprocessing as mp
from tqdm import tqdm
import warnings
from typing import Optional
import json
class HighQualityImageResizer:
"""High quality image resizer with sequential output names."""
def __init__(self,
input_dir: str,
output_dir: str,
max_size: int = 1920,
target_width: Optional[int] = None,
target_height: Optional[int] = None,
quality: int = 100,
interpolation: str = 'lanczos',
preserve_metadata: bool = True,
num_workers: int = 4,
format: str = 'jpg',
progressive: bool = True,
chroma_subsampling: int = 0,
optimize: bool = True,
start_id: int = 110,
prefix: str = "DJI_",
id_width: int = 5):
self.input_dir = Path(input_dir)
self.output_dir = Path(output_dir)
self.max_size = max_size
self.target_width = target_width
self.target_height = target_height
self.quality = min(max(quality, 1), 100)
self.preserve_metadata = preserve_metadata
self.num_workers = num_workers
self.format = format.lower()
self.progressive = progressive
self.chroma_subsampling = chroma_subsampling
self.optimize = optimize
# 鉁?缂栧彿绯荤粺
self.start_id = start_id
self.prefix = prefix
self.id_width = id_width
# 鉁?澶氳繘绋嬪畨鍏ㄨ鏁板櫒
self.id_lock = mp.Lock()
self.current_id = mp.Value('i', start_id)
self.interpolation_map = {
'nearest': cv2.INTER_NEAREST,
'linear': cv2.INTER_LINEAR,
'cubic': cv2.INTER_CUBIC,
'area': cv2.INTER_AREA,
'lanczos': cv2.INTER_LANCZOS4
}
self.interpolation = self.interpolation_map.get(interpolation, cv2.INTER_LANCZOS4)
self.output_dir.mkdir(parents=True, exist_ok=True)
self.log_file = self.output_dir / 'resize_log.json'
self.stats = {
'processed': 0,
'skipped': 0,
'failed': 0,
'files': []
}
# -------------------------
# 缂栧彿鐢熸垚锛堝杩涚▼瀹夊叏锛? # -------------------------
def _generate_output_name(self) -> str:
with self.id_lock:
current = self.current_id.value
self.current_id.value += 1
return f"{self.prefix}{current:0{self.id_width}d}.{self.format}"
# -------------------------
# EXIF 璇诲彇
# -------------------------
def _get_exif_metadata(self, image_path: Path):
metadata = {}
exif_bytes = None
try:
with Image.open(image_path) as img:
if 'exif' in img.info:
exif_dict = piexif.load(img.info['exif'])
exif_bytes = piexif.dump(exif_dict)
if '0th' in exif_dict:
metadata['camera_make'] = exif_dict['0th'].get(piexif.ImageIFD.Make, b'').decode(errors='ignore')
metadata['camera_model'] = exif_dict['0th'].get(piexif.ImageIFD.Model, b'').decode(errors='ignore')
metadata['date_time'] = exif_dict['0th'].get(piexif.ImageIFD.DateTime, b'').decode(errors='ignore')
metadata['format'] = img.format
metadata['mode'] = img.mode
metadata['size'] = img.size
except Exception as e:
warnings.warn(f"璇诲彇 {image_path} 鍏冩暟鎹け璐? {e}")
return exif_bytes, metadata
def _get_xmp_app1_segments(self, image_path: Path):
"""Return JPEG APP1 XMP segments, including DJI RTK metadata."""
try:
data = image_path.read_bytes()
except Exception as e:
warnings.warn(f"璇诲彇 XMP 澶辫触 {image_path}: {e}")
return []
if not data.startswith(b"\xff\xd8"):
return []
segments = []
pos = 2
while pos + 4 <= len(data):
if data[pos] != 0xFF:
break
while pos < len(data) and data[pos] == 0xFF:
pos += 1
if pos >= len(data):
break
marker = data[pos]
pos += 1
if marker in (0xD9, 0xDA):
break
if marker == 0x01 or 0xD0 <= marker <= 0xD7:
continue
if pos + 2 > len(data):
break
length = int.from_bytes(data[pos:pos + 2], "big")
if length < 2 or pos + length > len(data):
break
payload = data[pos + 2:pos + length]
segment = data[pos - 2:pos + length]
is_xmp = (
marker == 0xE1
and (
payload.startswith(b"http://ns.adobe.com/xap/1.0/\x00")
or b"drone-dji:" in payload
or b"RtkFlag" in payload
or b"RtkStdLat" in payload
)
)
if is_xmp:
segments.append(segment)
pos += length
return segments
def _insert_jpeg_app_segments(self, output_path: Path, segments):
if not segments:
return
try:
data = output_path.read_bytes()
if not data.startswith(b"\xff\xd8"):
return
if b"drone-dji:" in data or b"RtkFlag" in data:
return
output_path.write_bytes(data[:2] + b"".join(segments) + data[2:])
except Exception as e:
warnings.warn(f"鍐欏叆 XMP 澶辫触 {output_path}: {e}")
# -------------------------
# 灏哄璁＄畻
# -------------------------
def _calculate_target_size(self, width: int, height: int):
if self.target_width is not None and self.target_height is not None:
new_w = self.target_width
new_h = self.target_height
return new_w, new_h
if self.target_width is not None:
scale = self.target_width / width
new_w = self.target_width
new_h = int(height * scale)
# if new_w % 2 != 0:
# new_w += 1
# if new_h % 2 != 0:
# new_h += 1
return new_w, new_h
if width <= self.max_size and height <= self.max_size:
return width, height
if width > height:
scale = self.max_size / width
else:
scale = self.max_size / height
new_w = int(width * scale)
new_h = int(height * scale)
# if new_w % 2 != 0:
# new_w += 1
# if new_h % 2 != 0:
# new_h += 1
return new_w, new_h
# -------------------------
# 閿愬寲
# -------------------------
def _sharpen_image(self, image: np.ndarray, strength: float = 0.2):
if strength <= 0:
return image
kernel = np.array([[-strength, -strength, -strength],
[-strength, 1 + 8 * strength, -strength],
[-strength, -strength, -strength]])
sharpened = cv2.filter2D(image, -1, kernel)
return np.clip(sharpened, 0, 255).astype(np.uint8)
# -------------------------
# 淇濆瓨鍥惧儚
# -------------------------
def _encode_image(self, image: np.ndarray, output_path: Path, exif_bytes, xmp_segments=None):
try:
if len(image.shape) == 3 and image.shape[2] == 3:
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGR2RGB)
elif len(image.shape) == 3 and image.shape[2] == 4:
image_rgb = cv2.cvtColor(image, cv2.COLOR_BGRA2RGBA)
else:
image_rgb = image
pil_image = Image.fromarray(image_rgb)
if self.format in ('jpg', 'jpeg'):
attempts = [{
'quality': self.quality,
'optimize': self.optimize,
'progressive': self.progressive,
'subsampling': self.chroma_subsampling
}]
if exif_bytes and self.preserve_metadata:
attempts[0]['exif'] = exif_bytes
# Fallback for images that fail with optimize/progressive metadata saving.
attempts.append({
'quality': self.quality,
'optimize': False,
'progressive': False,
'subsampling': self.chroma_subsampling,
**({'exif': exif_bytes} if exif_bytes and self.preserve_metadata else {})
})
last_error = None
for save_kwargs in attempts:
try:
pil_image.save(output_path, **save_kwargs)
if self.preserve_metadata:
self._insert_jpeg_app_segments(output_path, xmp_segments or [])
return True, None
except Exception as e:
last_error = str(e)
return False, last_error
if self.format == 'png':
pil_image.save(output_path, compress_level=0)
return True, None
if self.format == 'webp':
pil_image.save(
output_path,
quality=self.quality,
method=6,
lossless=(self.quality == 100)
)
return True, None
pil_image.save(output_path)
return True, None
except Exception as e:
return False, str(e)
# -------------------------
# 鍗曞紶澶勭悊
# -------------------------
def _process_single_image(self, input_path: Path, output_path: Path):
result = {
'input': str(input_path),
'output': str(output_path),
'success': False
}
try:
exif_bytes, metadata = self._get_exif_metadata(input_path)
xmp_segments = self._get_xmp_app1_segments(input_path) if self.preserve_metadata else []
result['xmp_segments'] = len(xmp_segments)
# cv2.imread often fails on Windows paths containing Chinese characters.
image_data = np.fromfile(str(input_path), dtype=np.uint8)
image = cv2.imdecode(image_data, cv2.IMREAD_UNCHANGED)
if image is None:
result['error'] = '鏃犳硶璇诲彇鍥惧儚'
return result
h, w = image.shape[:2]
new_w, new_h = self._calculate_target_size(w, h)
if (w, h) != (new_w, new_h):
image = cv2.resize(image, (new_w, new_h), interpolation=self.interpolation)
image = self._sharpen_image(image)
success, save_error = self._encode_image(image, output_path, exif_bytes, xmp_segments)
result['success'] = success
if save_error:
result['error'] = save_error
warnings.warn(f"淇濆瓨澶辫触 {output_path}: {save_error}")
except Exception as e:
result['error'] = str(e)
return result
# -------------------------
# Worker
# -------------------------
def _process_worker(self, file_queue: mp.Queue, result_queue: mp.Queue):
while True:
task = file_queue.get()
if task is None:
break
input_path, output_path = task
result = self._process_single_image(input_path, output_path)
result_queue.put(result)
# -------------------------
# 鏂囦欢鎵弿
# -------------------------
def _get_image_files(self):
exts = {'.jpg', '.jpeg', '.png', '.webp', '.tiff', '.tif', '.bmp'}
files = {}
for ext in exts:
for path in self.input_dir.glob(f'*{ext}'):
files[str(path).lower()] = path
for path in self.input_dir.glob(f'*{ext.upper()}'):
files[str(path).lower()] = path
return sorted(files.values(), key=lambda path: path.name.lower())
# -------------------------
# 涓绘祦绋? # -------------------------
def process(self):
image_files = self._get_image_files()
if not image_files:
print("No image files found")
return
print(f"Found {len(image_files)} images")
file_queue = mp.Queue()
result_queue = mp.Queue()
for img_file in image_files:
output_name = self._generate_output_name()
output_path = self.output_dir / output_name
file_queue.put((img_file, output_path))
for _ in range(self.num_workers):
file_queue.put(None)
processes = []
for _ in range(self.num_workers):
p = mp.Process(target=self._process_worker, args=(file_queue, result_queue))
p.start()
processes.append(p)
with tqdm(total=len(image_files), desc="澶勭悊杩涘害") as pbar:
for _ in range(len(image_files)):
result = result_queue.get()
self.stats['files'].append(result)
if result['success']:
self.stats['processed'] += 1
else:
self.stats['failed'] += 1
pbar.update(1)
for p in processes:
p.join()
with open(self.log_file, 'w', encoding='utf-8') as f:
json.dump(self.stats, f, indent=2, ensure_ascii=False)
print("\n鉁?瀹屾垚")
print(f"鎴愬姛: {self.stats['processed']}")
print(f"澶辫触: {self.stats['failed']}")
# ============================
# CLI
# Default run:
# python tool1_Resize_img_SetID.py
# Optional override:
# python tool1_Resize_img_SetID.py INPUT_DIR OUTPUT_DIR --resolution 1805 1203 --start-id 0
# ============================
def main():
default_input_dir = r"D:\0ubuntu20\1slam\数据\2RTK\City4_outskirts2\map_cloudy_0303_15pm_135m\images_source"
default_output_dir = r"D:\0ubuntu20\1slam\数据\2RTK\City4_outskirts2\map_cloudy_0303_15pm_135m\images"
default_resolution = (1805, 1203)
default_start_id = 0
parser = argparse.ArgumentParser(description="High quality image resize and sequential rename")
parser.add_argument("input_dir", nargs="?", default=default_input_dir)
parser.add_argument("output_dir", nargs="?", default=default_output_dir)
parser.add_argument("-s", "--size", type=int, default=1805,
help="Limit by longest side, default 1805")
parser.add_argument("--width", type=int, default=None,
help="Set output width only; height follows original aspect ratio")
parser.add_argument("--resolution", nargs=2, type=int, metavar=("WIDTH", "HEIGHT"), default=default_resolution,
help="Set exact output resolution, for example --resolution 1920 1080")
parser.add_argument("-q", "--quality", type=int, default=100)
parser.add_argument("-w", "--workers", type=int, default=8)
parser.add_argument("--start-id", type=int, default=default_start_id)
parser.add_argument("--prefix", type=str, default="DJI_")
parser.add_argument("--id-width", type=int, default=5)
args = parser.parse_args()
if args.width is not None and args.width <= 0:
parser.error("--width 蹇呴』澶т簬 0")
if args.resolution is not None:
if args.width is not None:
parser.error("--resolution 鍜?--width 涓嶈兘鍚屾椂浣跨敤")
if args.resolution[0] <= 0 or args.resolution[1] <= 0:
parser.error("--resolution 鐨勫鍜岄珮閮藉繀椤诲ぇ浜?0")
target_width = None
target_height = None
if args.resolution is not None:
target_width, target_height = args.resolution
elif args.width is not None:
target_width = args.width
resizer = HighQualityImageResizer(
input_dir=args.input_dir,
output_dir=args.output_dir,
max_size=args.size,
target_width=target_width,
target_height=target_height,
quality=args.quality,
num_workers=args.workers,
start_id=args.start_id,
prefix=args.prefix,
id_width=args.id_width
)
resizer.process()
if __name__ == "__main__":
main()
浙公网安备 33010602011771号