ByoungSooPark/BlenderLLM
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update README.md

Browse Source
This commit is contained in:
Yuhao Du
2024-12-19 21:13:07 +08:00
parent 8b68495e13
commit 7d67662831
4 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -24,7 +24,7 @@ Welcome to the repository of **BlenderLLM**. **BlenderLLM** is a large language
Here is a list of what has been released: Here is a list of what has been released:
1. **BlendNet**: A high-quality dataset containing $8k$ `<instruction, CAD script>` pairs. 1. **BlendNet**: A high-quality dataset containing $12k$ `<instruction, CAD script>` pairs.
2. **BlenderLLM**: A large language model fine-tuned on BlendNet based on **Qwen2.5-Coder-7B-Instruct**, designed to output CAD scripts. 2. **BlenderLLM**: A large language model fine-tuned on BlendNet based on **Qwen2.5-Coder-7B-Instruct**, designed to output CAD scripts.
3. **CADBench**: A comprehensive benchmark for evaluating this task. 3. **CADBench**: A comprehensive benchmark for evaluating this task.

7
modeling.py
View File

@ -2,7 +2,7 @@ import os
import argparse import argparse
from scripts.infer import generate_response from scripts.infer import generate_response
from scripts.blender_runner import run_blender_script from scripts.blender_runner import run_blender_script
from scripts.geometry_utils import parse_obj_file, calculate_bounding_box from scripts.geometry_utils import calculate_bounding_box
from scripts.config import CAMERA_ANGLES, BRIGHTNESS from scripts.config import CAMERA_ANGLES, BRIGHTNESS
def generate_blender_script(model_name, prompt): def generate_blender_script(model_name, prompt):
@ -25,8 +25,7 @@ def run_script_and_save_obj(script, obj_name, output_folder, blender_executable)
def calculate_and_render_image(script, obj_name, output_folder, obj_path, blender_executable, brightness): def calculate_and_render_image(script, obj_name, output_folder, obj_path, blender_executable, brightness):
"""Calculate bounding box and render the image using Blender script.""" """Calculate bounding box and render the image using Blender script."""
ensure_output_folder_exists(output_folder) ensure_output_folder_exists(output_folder)
vertices = parse_obj_file(obj_path) bounding_coords = calculate_bounding_box(obj_path)
bounding_coords = calculate_bounding_box(vertices)
brightness_value = BRIGHTNESS.get(brightness, BRIGHTNESS["Very Dark"]) brightness_value = BRIGHTNESS.get(brightness, BRIGHTNESS["Very Dark"])
run_blender_script( run_blender_script(
script, script,
@ -47,7 +46,7 @@ def parse_arguments():
parser.add_argument("--obj_name", type=str, default="cube", help="Name of the generated object file.") parser.add_argument("--obj_name", type=str, default="cube", help="Name of the generated object file.")
parser.add_argument("--output_folder", type=str, default="images/cube", help="Folder to save output files.") parser.add_argument("--output_folder", type=str, default="images/cube", help="Folder to save output files.")
parser.add_argument("--blender_executable", type=str, default="blender", help="Path to Blender executable.") parser.add_argument("--blender_executable", type=str, default="blender", help="Path to Blender executable.")
parser.add_argument("--brightness", type=str, default="Dark", choices=BRIGHTNESS.keys(), help="Brightness level for the rendered image. Options: Very Bright, Bright, Medium Bright, Dark, Very Dark.") parser.add_argument("--brightness", type=str, default="Very Dark", choices=BRIGHTNESS.keys(), help="Brightness level for the rendered image. Options: Very Bright, Bright, Medium Bright, Dark, Very Dark.")
return parser.parse_args() return parser.parse_args()
def main(): def main():

1
requirements.txt
View File

@ -2,3 +2,4 @@ numpy
torch torch
git+https://github.com/huggingface/transformers.git git+https://github.com/huggingface/transformers.git
sentencepiece sentencepiece
accelerate>=0.26.0

12
scripts/config.py
View File

@ -1,14 +1,14 @@
import math import math
CAMERA_ANGLES = [ CAMERA_ANGLES = [
(math.radians(45), 0, math.radians(45)), (math.radians(315), math.radians(180), math.radians(135)),
(math.radians(45), 0, math.radians(135)),
(math.radians(45), 0, math.radians(225)),
(math.radians(45), 0, math.radians(315)), (math.radians(45), 0, math.radians(315)),
(math.radians(315), math.radians(180), math.radians(225)),
(math.radians(315), math.radians(180), math.radians(315)),
(math.radians(315), math.radians(180), math.radians(45)), (math.radians(315), math.radians(180), math.radians(45)),
(math.radians(315), math.radians(180), math.radians(135)) (math.radians(45), 0, math.radians(225)),
(math.radians(315), math.radians(180), math.radians(225)),
(math.radians(45), 0, math.radians(45)),
(math.radians(315), math.radians(180), math.radians(315)),
(math.radians(45), 0, math.radians(135))
] ]