You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
179 lines
8.4 KiB
179 lines
8.4 KiB
import os, sys, time, shutil, subprocess, shlex, json,oss2 |
|
from PIL import Image |
|
import platform |
|
if platform.system() == 'Windows': |
|
sys.path.append('e:\\libs\\') |
|
else: |
|
sys.path.append('/data/deploy/make3d/make2/libs/') |
|
|
|
import config, libs |
|
|
|
#检测是否存在 rate_xmps 文件,如果没有则要创建该影棚对应的rate_xmps文件 |
|
def check_rate_xmps(pid): |
|
pid = str(pid) |
|
psid = libs.getPSid(pid) |
|
#判断是否存在该目录 |
|
if config.oss_bucket.object_exists(f'xmps_new_studio_rate/{psid}/') == False: |
|
return False |
|
return True |
|
|
|
def clearExifAndRotate(imagePath, rotate=90): |
|
if os.path.exists(imagePath) == False: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {imagePath} 图片不存在') |
|
return |
|
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {imagePath} 图片exif信息清除并且旋转...') |
|
# 打开图像 |
|
with Image.open(imagePath) as img: |
|
# 使用RGB模式打开图像来丢弃可能存在的EXIF信息 |
|
img = img.convert("RGB") |
|
img = img.rotate(rotate,expand=True) |
|
# 保存图像，保存时不会带有原始的任何元数据 |
|
img.save(imagePath, quality=100) # 对于JPEG，可以调整质量参数 |
|
|
|
#如果老影棚没有做过旋转坐标的照片的，则要做一次旋转坐标 |
|
def rateImages(pid): |
|
#遍历文件夹下的所有图片 |
|
# photo1_path = os.path.join(config.workdir, pid, 'photo1') |
|
photo2_path = os.path.join(config.workdir, pid, 'photo2') |
|
# for pic in os.listdir(photo1_path): |
|
# if pic.endswith('.jpg'): |
|
# #清除图片exif信息， 旋转图片 180度 |
|
# clearExifAndRotate(os.path.join(photo1_path, pic)) |
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pic} 图片exif信息清除并且旋转完成') |
|
|
|
for pic in os.listdir(photo2_path): |
|
if pic.endswith('.jpg'): |
|
#判断文件大小 |
|
if os.path.getsize(os.path.join(photo2_path, pic)) < 1024 * 1024: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pic} 图片大小小于1M，不做处理') |
|
#删除小于1M的图片 |
|
os.remove(os.path.join(photo2_path, pic)) |
|
continue |
|
#清除图片exif信息， 旋转图片 180度 |
|
clearExifAndRotate(os.path.join(photo2_path, pic)) |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pic} 图片exif信息清除并且旋转完成') |
|
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} 图片exif信息清除并且旋转完成') |
|
|
|
|
|
def upload_xmp(pid): |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} 上传xmp文件之前先删除oss上的xmps_rate文件所在目录...') |
|
pid = str(pid) |
|
psid = libs.getPSid(pid) |
|
#移除掉旧的文件夹 |
|
#config.oss_bucket.delete_object(f'xmps/{pid}/') |
|
#删除oss 上的文件夹里的内容 |
|
|
|
#判断是否存在该目录 |
|
if config.oss_bucket.object_exists(f'xmps_new_studio_rate/{psid}/') == True: |
|
object_list = oss2.ObjectIterator(config.oss_bucket, prefix=f'xmps_new_studio_rate/{psid}/') |
|
if not any(object_list): |
|
config.oss_bucket.batch_delete_objects([obj.key for obj in object_list]) |
|
|
|
start_time = time.time() |
|
workdir = os.path.join(config.workdir, pid) |
|
|
|
config.oss_bucket.put_object_from_file(f'xmps_new_studio_rate/{psid}/{psid}.rcbox', os.path.join(workdir, f'{pid}.rcbox')) |
|
|
|
# for xmp in os.listdir(os.path.join(workdir, 'photo1')): |
|
# if xmp.endswith('.xmp'): |
|
# config.oss_bucket.put_object_from_file(f'xmps_new_studio_rate/{psid}/mesh/{xmp}', os.path.join(workdir, 'photo1', xmp)) |
|
# #复制一份到共享盘上的 libs xmps psid 文件夹下 |
|
# #判断是否有对应的文件夹,不存在就创建 |
|
# os.makedirs(os.path.join(config.sharedir, 'xmps_new_studio_rate', psid, 'mesh'), exist_ok=True) |
|
|
|
# shutil.copy(os.path.join(workdir, 'photo1', xmp), os.path.join(config.sharedir, 'xmps_new_studio_rate', psid, 'mesh', xmp)) |
|
|
|
for xmp in os.listdir(os.path.join(workdir, 'photo2')): |
|
if xmp.endswith('.xmp'): |
|
config.oss_bucket.put_object_from_file(f'xmps_new_studio_rate/{psid}/texture/{xmp}', os.path.join(workdir, 'photo2', xmp)) |
|
#复制一份到共享盘上的 libs xmps psid 文件夹下 |
|
os.makedirs(os.path.join(config.sharedir, 'xmps_new_studio_rate', psid, 'texture'), exist_ok=True) |
|
shutil.copy(os.path.join(workdir, 'photo2', xmp), os.path.join(config.sharedir, 'xmps_new_studio_rate', psid, 'texture', xmp)) |
|
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} pid: {pid} xmp文件上传完成，共费时{time.time() - start_time}秒') |
|
|
|
|
|
#创建rate_xmps文件 |
|
def create_rate_xmps(pid): |
|
start_time = time.time() |
|
pid = str(pid) |
|
psid = libs.getPSid(pid) |
|
#判断是否存在xmps_rate文件夹 |
|
flag = check_rate_xmps(pid) |
|
if flag == True: |
|
# 提示输入指令是否要重新做rate_xmps文件 |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} xmps_new_studio_rate文件已经存在，是否重新生成rate_xmps文件？') |
|
continue_or_not = input('是否重新生成xmps_new_studio_rate文件？(y/n)') |
|
if continue_or_not == 'y': |
|
pass |
|
else: |
|
sys.exit(0) |
|
|
|
#判断是否有对应的 photo1 和 photo2 文件夹 |
|
# photo1_path = os.path.join(config.workdir, pid, 'photo1') |
|
photo2_path = os.path.join(config.workdir, pid, 'photo2') |
|
|
|
#判断是否存在目录 |
|
if os.path.exists(photo2_path) == False: |
|
#重新下载图片文件photo1和photo2 |
|
libs.down_from_oss(config.oss_bucket, config.workdir, pid) |
|
|
|
#photos1_count = len(os.listdir(photo1_path)) |
|
photos2_count = len(os.listdir(photo2_path)) |
|
if photos2_count < 21: |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} photo2数量{photos2_count}，未能覆盖所有相机，是否继续计算相机位姿？') |
|
continue_or_not = input('是否继续计算相机位姿？(y/n)') |
|
if continue_or_not == 'y': |
|
pass |
|
else: |
|
sys.exit(0) |
|
|
|
#清除图片exif信息， 旋转图片 180度 |
|
rateImages(pid) |
|
#通过rc 计算出 xmps 文件 |
|
exportxmp = ' -exportXMP "D:\\make2\\config\\exportXMP.config.xml" ' |
|
cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" \ |
|
-addFolder "{os.path.join(config.workdir, pid, "photo2")}" -selectAllImages \ |
|
-align -align -align -align -align \ |
|
{exportxmp} \ |
|
-save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" -quit' |
|
print(cmd) |
|
cmd = shlex.split(cmd) |
|
res = subprocess.run(cmd) |
|
# TODO：加入report相机位姿质量评估 |
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} photo1相机位姿完成，共费时{libs.diff_time(start_time)}') |
|
|
|
# for xmp in os.listdir(photo1_path): |
|
# if xmp.endswith('.xmp'): |
|
# shutil.copy(os.path.join(photo1_path, xmp), os.path.join(photo2_path, xmp.replace('_1.xmp', '_8.xmp'))) |
|
|
|
# psid = libs.getPSid(pid) |
|
# cmd = f'{config.rcbin} {config.r2["init"]} -setInstanceName {pid} \ |
|
# -load "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}" {config.r["setTextureFalse"]} \ |
|
# -addFolder "{os.path.join(config.workdir, pid, "photo2")}" -selectAllImages \ |
|
# -detectMarkers "D:\\make2\\config\\detectMarkers.config.xml" \ |
|
# {libs.get_defineDistances(psid)} -update -align -align {config.r2["setRegion"]} \ |
|
# {exportxmp} \ |
|
# -exportReconstructionRegion "{os.path.join(config.workdir, pid, f"{pid}.rcbox")}" \ |
|
# -save "{os.path.join(config.workdir, pid, f"{pid}.rcproj")}"' |
|
# print(cmd) |
|
# cmd = shlex.split(cmd) |
|
# res = subprocess.run(cmd) |
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} photo2相机位姿完成，共费时{libs.diff_time(start_time)}') |
|
|
|
|
|
upload_or_not = input('是否上传oss？(y/n)') |
|
if upload_or_not == 'y': |
|
upload_xmp(pid) |
|
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())} {pid} rate_xmps文件生成完成，共费时{libs.diff_time(start_time)}') |
|
|
|
|
|
if __name__ == '__main__': |
|
pid = sys.argv[1] |
|
print(f'执行任务的pid: {pid}') |
|
create_rate_xmps(pid) |
|
|
|
|