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

test different art styles, change pipeline prompt

Browse Source
This commit is contained in:
henrisel
2025-12-05 16:58:07 +02:00
parent 408949e5c2
commit 843bd141eb
4 changed files with 35 additions and 54 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
3d-generation-pipeline/.gitignore vendored
View File

@@ -3,4 +3,5 @@
__pycache__ __pycache__
images/ images/
models/ models/
logs/ logs/
notebooks/test_resources/

13
3d-generation-pipeline/notebooks/cloudflare_API_test.ipynb
View File

@@ -2,7 +2,7 @@
"cells": [ "cells": [
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": 1,
"id": "1dc6faae", "id": "1dc6faae",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -15,7 +15,7 @@
}, },
{ {
"cell_type": "code", "cell_type": "code",
"execution_count": null, "execution_count": 2,
"id": "b3107275", "id": "b3107275",
"metadata": {}, "metadata": {},
"outputs": [], "outputs": [],
@@ -44,17 +44,18 @@
"name": "stdout", "name": "stdout",
"output_type": "stream", "output_type": "stream",
"text": [ "text": [
"Saved: output.jpg (263282 bytes)\n", "Saved: test_resources/resolution_test_1.jpg (315728 bytes)\n"
"Saved: image9.jpg (263282 bytes)\n"
] ]
} }
], ],
"source": [ "source": [
"# https://developers.cloudflare.com/workers-ai/models/flux-1-schnell/\n",
"\n",
"MODEL = \"@cf/black-forest-labs/flux-1-schnell\"\n", "MODEL = \"@cf/black-forest-labs/flux-1-schnell\"\n",
"URL = f\"https://api.cloudflare.com/client/v4/accounts/{ACCOUNT_ID}/ai/run/{MODEL}\"\n", "URL = f\"https://api.cloudflare.com/client/v4/accounts/{ACCOUNT_ID}/ai/run/{MODEL}\"\n",
"\n", "\n",
"payload = {\n", "payload = {\n",
" \"prompt\": \"a slightly curved broadsword with a fancy golden crossguard\",\n", " \"prompt\": \"cyborg crocodile, realistic style, single object, front and side fully visible, plain neutral background, clear details, soft studio lighting, true-to-scale\",\n",
"}\n", "}\n",
"\n", "\n",
"headers = {\n", "headers = {\n",
@@ -72,7 +73,7 @@
"\n", "\n",
"img_bytes = base64.b64decode(b64)\n", "img_bytes = base64.b64decode(b64)\n",
"\n", "\n",
"out_path = \"output.jpg\"\n", "out_path = \"test_resources/resolution_test_1.jpg\"\n",
"with open(out_path, \"wb\") as f:\n", "with open(out_path, \"wb\") as f:\n",
" f.write(img_bytes)\n", " f.write(img_bytes)\n",
"\n", "\n",

71
3d-generation-pipeline/notebooks/local_model_generation_test.ipynb
View File

File diff suppressed because one or more lines are too long

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

@@ -33,7 +33,7 @@ async def main():
input_prompt = args.prompt input_prompt = args.prompt
print(f"Input prompt: {input_prompt}") print(f"Input prompt: {input_prompt}")
image_generation_prompt = input_prompt + ", single object, fully visible, plain neutral background, high-resolution, sharp details, soft studio lighting, front and side visible, realistic style, true-to-scale" image_generation_prompt = input_prompt + ", realistic style, single object, front and side fully visible, plain neutral background, clear details, soft studio lighting, true-to-scale"
pipeline_folder = Path(__file__).resolve().parent pipeline_folder = Path(__file__).resolve().parent
timestamp = get_timestamp() timestamp = get_timestamp()