157 lines
7.4 KiB
Python
157 lines
7.4 KiB
Python
import os, sys, time, argparse, requests, json, re, oss2
|
|
import bpy, bmesh
|
|
import open3d as o3d
|
|
import numpy as np
|
|
# import cupy as cp
|
|
from mathutils import Matrix
|
|
import math
|
|
import logging
|
|
import matplotlib.pyplot as plt
|
|
|
|
|
|
def get_auto_obj_bake_vertex_ply(obj_filepath, save_ply_path):
|
|
start_time = time.time()
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: Getting ply vertex from {obj_filepath}')
|
|
bpy.ops.object.select_all(action='DESELECT')
|
|
bpy.ops.object.select_by_type(type='MESH')
|
|
bpy.ops.object.delete(use_global=False, confirm=False)
|
|
bpy.ops.wm.obj_import(filepath=obj_filepath)
|
|
|
|
obj_original = [obj for obj in bpy.data.objects if obj.type == 'MESH'][0]
|
|
bpy.context.view_layer.objects.active = obj_original
|
|
obj_original.select_set(True)
|
|
pid = obj_original.name
|
|
|
|
obj_num_vertices = len(obj_original.data.vertices)
|
|
# print(f"模型 {pid} 顶点数量:", len(obj_original.data.vertices))
|
|
|
|
# 选择新分离出来的网格对象，并将该对象设为进活动对象
|
|
bpy.ops.object.select_all(action='DESELECT')
|
|
bpy.context.view_layer.objects.active = obj_original
|
|
obj_original.select_set(True)
|
|
|
|
# 添加颜色属性
|
|
bpy.ops.geometry.color_attribute_add()
|
|
# 节点的创建：
|
|
nodes = bpy.context.active_object.active_material.node_tree.nodes
|
|
links = bpy.context.active_object.active_material.node_tree.links
|
|
tex_coord = nodes.new("ShaderNodeTexCoord")
|
|
Mapping = nodes.new('ShaderNodeMapping')
|
|
vertex_color = nodes.new("ShaderNodeVertexColor")
|
|
tex_coord.location = (-800, 600)
|
|
Mapping.location = (-500, 200)
|
|
vertex_color.location = (-200, 600)
|
|
|
|
# 节点之间的连接：
|
|
# 纹理坐标的UV 连接 映射的矢量；映射的矢量 连接 纹理贴图的矢量；原理化BSDF删除；纹理贴图的矢量 连接 表（曲）面。
|
|
links.new(tex_coord.outputs[2], nodes['Mapping'].inputs[0])
|
|
links.new(Mapping.outputs[0], nodes['Image Texture'].inputs[0])
|
|
principled_bsdf = nodes.get("Principled BSDF")
|
|
nodes.remove(principled_bsdf)
|
|
links.new(nodes['Image Texture'].outputs[0], nodes['Material Output'].inputs[0])
|
|
|
|
# 4.渲染烘焙相关操作
|
|
bpy.context.scene.render.engine = 'CYCLES'
|
|
bpy.context.scene.cycles.device = 'GPU'
|
|
bpy.context.scene.cycles.preview_samples = 1
|
|
bpy.context.scene.cycles.samples = 1
|
|
bpy.context.scene.cycles.bake_type = 'EMIT'
|
|
bpy.context.scene.render.bake.use_selected_to_active = False
|
|
bpy.context.scene.render.bake.target = 'VERTEX_COLORS'
|
|
bpy.ops.object.bake(type='EMIT') # 开始 Bake
|
|
|
|
# 连接节点 Color Attribute 和 节点 Material Output
|
|
links.new(nodes['Color Attribute'].outputs[0], nodes['Material Output'].inputs[0])
|
|
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: Bake EMIT time: ', time.time() - start_time)
|
|
start_time = time.time()
|
|
|
|
# 5.输出模型
|
|
# bpy.ops.wm.save_as_mainfile(filepath='/data/datasets/plys/121614.bend')
|
|
bpy.ops.wm.ply_export(filepath=save_ply_path, apply_modifiers=True, export_selected_objects=True, export_uv=False,
|
|
export_normals=False, export_colors='SRGB', export_triangulated_mesh=False, ascii_format=True)
|
|
bpy.data.objects.remove(obj_original, do_unlink=True)
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: Save ply time: ', time.time() - start_time)
|
|
|
|
return obj_num_vertices
|
|
|
|
|
|
|
|
if __name__ == '__main__':
|
|
# workdir = '/data/datasets/suwa_cut/data/auto/'
|
|
# output = '/data/datasets/plys/auto/'
|
|
workdir = 'G:\obj_fix\print'
|
|
output = 'G:\obj_fix\ply\whole_obj_bake_vertex_high_ply'
|
|
directory = f'G:\obj_fix/print/right'
|
|
#遍历获取 G:\obj_fix\print\right 目录的文件，根据文件名获取 pid
|
|
for file_name in os.listdir(directory):
|
|
# 获取文件的完整路径
|
|
file_path = os.path.join(directory, file_name)
|
|
# 检查是否是文件
|
|
if os.path.isfile(file_path):
|
|
#
|
|
arrFileName = file_name.split('_')
|
|
pid = arrFileName[0]
|
|
if int(pid) == 0:
|
|
continue
|
|
|
|
print(f"File: {file_name}, PID: {pid}")
|
|
|
|
#判断文件是否存在
|
|
file_path = os.path.join(output, f'{pid}.ply')
|
|
if os.path.exists(file_path) and os.path.isfile(file_path):
|
|
print("文件已存在，跳过循环！！！")
|
|
continue
|
|
|
|
obj_filepath = os.path.join(workdir, f'{pid}/{pid}.obj')
|
|
save_ply_path = os.path.join(output, f'{pid}.ply')
|
|
if not os.path.exists(obj_filepath):
|
|
print(f"{obj_filepath} 文件不存在，跳过循环！！！")
|
|
continue
|
|
|
|
obj_num_vertices = get_auto_obj_bake_vertex_ply(obj_filepath, save_ply_path)
|
|
if obj_num_vertices == 0:
|
|
print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: 模型 {pid} 程序getHeadCount()无法找到模型人数！总共运行时间: ', time.time() - start_time0)
|
|
logging.info(f'模型 {pid} 顶点数量: {obj_num_vertices}，程序getHeadCount()无法找到模型人数！，程序总共运行时间: {time.time() - start_time0:.2f} 秒')
|
|
continue
|
|
# # 从文件名中提取 pid
|
|
# pid = extract_pid(file_name)
|
|
# if pid:
|
|
# print(f"File: {file_name}, PID: {pid}")
|
|
# else:
|
|
# print(f"Unable to extract PID from file: {file_name}")
|
|
|
|
|
|
# return
|
|
# listdir_pids = os.listdir(workdir)
|
|
# for pid in listdir_pids:
|
|
# file_path = os.path.join(output, f'{pid}/{pid}.ply')
|
|
# if os.path.exists(file_path) and os.path.isfile(file_path):
|
|
# print("文件已存在，跳过循环！！！")
|
|
# continue
|
|
|
|
# print(pid)
|
|
# logging.basicConfig(filename=f'/data/datasets/plys/auto_vertex_time.log', level=logging.INFO, format='%(asctime)s - %(levelname)s - %(message)s')
|
|
# start_time0 = time.time()
|
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: 模型 {pid} 开始运行时间: ', start_time0)
|
|
# # 判断输出目录是否存在
|
|
# if not os.path.exists(os.path.join(output, f'{pid}/')):
|
|
# os.makedirs(os.path.join(output, f'{pid}/'))
|
|
|
|
# obj_filepath = os.path.join(workdir, f'{pid}/{pid}_cpd.obj')
|
|
# save_ply_path = os.path.join(output, f'{pid}/{pid}.ply')
|
|
# if not os.path.exists(obj_filepath):
|
|
# print(f"{obj_filepath} 文件不存在，跳过循环！！！")
|
|
# continue
|
|
|
|
# obj_num_vertices = get_auto_obj_bake_vertex_ply(obj_filepath, save_ply_path)
|
|
# if obj_num_vertices == 0:
|
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: 模型 {pid} 程序getHeadCount()无法找到模型人数！总共运行时间: ', time.time() - start_time0)
|
|
# logging.info(f'模型 {pid} 顶点数量: {obj_num_vertices}，程序getHeadCount()无法找到模型人数！，程序总共运行时间: {time.time() - start_time0:.2f} 秒')
|
|
# continue
|
|
|
|
# print(f'{time.strftime("%Y-%m-%d %H:%M:%S", time.localtime())}: 模型 {pid} 顶点数量: {obj_num_vertices} 程序总共运行时间: ', time.time() - start_time0)
|
|
# logging.info(f'模型 {pid} 顶点数量: {obj_num_vertices}，程序总共运行时间: {time.time() - start_time0:.2f} 秒')
|
|
# # 退出 Blender 并释放资源
|
|
# bpy.ops.wm.quit_blender()
|
|
|