henrisel/DeltaVR3DModelGeneration
1
0
Fork 0
forked from cgvr/DeltaVR
Code Pull Requests Activity

small cleanup of pipeline script

Browse Source
This commit is contained in:
henrisel
2025-12-17 15:46:01 +02:00
parent 2c19602e9b
commit 409b9fbb51
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
3d-generation-pipeline/generate_model_local.py
View File

@@ -59,7 +59,7 @@ def generate_no_preview(image_base64: str):
print(f"Error: {str(e)}") print(f"Error: {str(e)}")
return None return None
def image_to_3d_api(image_path, output_path): def image_to_3d_trellis(image_path, output_path):
with open(image_path, 'rb') as image_file: with open(image_path, 'rb') as image_file:
image_data = image_file.read() image_data = image_file.read()

4
3d-generation-pipeline/start_pipeline.py
View File

@@ -8,7 +8,7 @@ from datetime import datetime
from dotenv import load_dotenv from dotenv import load_dotenv
from generate_image_local import text_to_image_invoke_ai from generate_image_local import text_to_image_invoke_ai
from generate_model_local import image_to_3d_api from generate_model_local import image_to_3d_trellis
load_dotenv() load_dotenv()
@@ -51,7 +51,7 @@ async def main():
print(f"Generated image file: {image_path}") print(f"Generated image file: {image_path}")
model_path = pipeline_folder / "models" / timestamp model_path = pipeline_folder / "models" / timestamp
model_file = image_to_3d_api(image_path, model_path) model_file = image_to_3d_trellis(image_path, model_path)
model_generation_time = time.time() - time_checkpoint model_generation_time = time.time() - time_checkpoint
logging.info(f"Model generation time: {round(model_generation_time, 1)} s") logging.info(f"Model generation time: {round(model_generation_time, 1)} s")