mirror of
https://github.com/nerfstudio-project/nerfstudio.git
synced 2026-11-12 19:04:03 +08:00
71 lines
2.8 KiB
Python
71 lines
2.8 KiB
Python
import os
|
|
import shutil
|
|
import re
|
|
import argparse
|
|
import subprocess
|
|
from pathlib import Path
|
|
|
|
|
|
def copy_files(input_dir):
|
|
brackets = os.listdir(input_dir)
|
|
|
|
target_exr = os.path.join(input_dir, "GT_exr")
|
|
target_jpg = os.path.join(input_dir, "GT_jpg")
|
|
target_jpg_resize = os.path.join(input_dir, "GT_jpg_resize")
|
|
os.makedirs(target_exr, exist_ok=True)
|
|
os.makedirs(target_jpg, exist_ok=True)
|
|
|
|
for folder in brackets:
|
|
if not folder.startswith("GT_"):
|
|
folder_path = os.path.join(input_dir, folder)
|
|
if os.path.isdir(folder_path):
|
|
smallest_value = None
|
|
smallest_file = None
|
|
for file in os.listdir(folder_path):
|
|
if file.startswith("GT") and file.lower().endswith(".exr"):
|
|
source_file = os.path.join(folder_path, file)
|
|
destination_file = os.path.join(target_exr, file)
|
|
shutil.copy2(source_file, destination_file)
|
|
print(f"Copied EXR: {source_file} -> {destination_file}")
|
|
|
|
if file.lower().endswith(".jpg"):
|
|
numbers = re.findall(r'\d+', file)
|
|
num = int(numbers[0])
|
|
if smallest_value is None or num < smallest_value:
|
|
smallest_value = num
|
|
smallest_file = file
|
|
|
|
source_file = os.path.join(folder_path, smallest_file)
|
|
destination_file = os.path.join(target_jpg, f"{folder}.jpg")
|
|
shutil.copy2(source_file, destination_file)
|
|
print(f"Copied JPG: {source_file} -> {destination_file}")
|
|
|
|
# Run a simple command
|
|
print("Copying target_jpg to target_jpg_resize...")
|
|
result = subprocess.run(["cp", "-r", target_jpg, target_jpg_resize], capture_output=True, text=True)
|
|
print(result)
|
|
|
|
print("Resizing target_jpg_resize...")
|
|
result = subprocess.run(["find", target_jpg_resize, "-iname", "*.jpg", "-exec", "convert", "{}", "-verbose", "-resize", "3840x1920>", "{}", ";"], capture_output=True, text=True)
|
|
print(result)
|
|
|
|
gt_path = Path(input_dir)
|
|
sfm_path = os.path.join(gt_path.parent, 'sfm/images')
|
|
for file in os.listdir(target_jpg_resize):
|
|
source_file = os.path.join(target_jpg_resize, file)
|
|
destination_file = os.path.join(sfm_path, file)
|
|
shutil.copy2(source_file, destination_file)
|
|
print(f"Copied JPG: {source_file} -> {destination_file}")
|
|
|
|
|
|
if __name__ == '__main__':
|
|
parser = argparse.ArgumentParser()
|
|
parser.add_argument('--input_dir', type=str, required=True,
|
|
help='GT directory containing all the subdirectories (e.g., GT1, GT2, ...) with frames')
|
|
args = parser.parse_args()
|
|
|
|
copy_files(args.input_dir)
|
|
|
|
|
|
|