23 lines
592 B
Python
23 lines
592 B
Python
import os
|
||
import sys
|
||
import shutil
|
||
|
||
folder_path = sys.argv[1]
|
||
|
||
colmap_path = "./colmap_tmp"
|
||
images_path = os.path.join(colmap_path, "images")
|
||
os.makedirs(images_path, exist_ok=True)
|
||
i=0
|
||
|
||
dir1=os.path.join("data",folder_path)
|
||
for folder_name in os.listdir(dir1):
|
||
dir2=os.path.join(dir1,folder_name)
|
||
for file_name in os.listdir(dir2):
|
||
if file_name.startswith("frame_00001"):
|
||
i=i+1
|
||
src_path = os.path.join(dir2, file_name)
|
||
dst_path = os.path.join(images_path, f"image{i}.jpg")
|
||
shutil.copyfile(src_path, dst_path)
|
||
|
||
print("End！")
|