Comment out KDTree functionality in generate3d function to simplify color assignment process. Updated mesh creation to set vertex colors to None, removing reliance on nearest neighbor color mapping.
Refactor generate3d function in inference.py to streamline mesh export process. Removed manual MTL file handling and updated export calls to enhance code clarity and maintainability.
Refactor generate3d function in inference.py to improve mesh export process. Introduced temporary file handling for OBJ, MTL, and PNG outputs, ensuring proper texture mapping. Enhanced code readability by restructuring renderer initialization and file writing logic.
Refactor device handling in generate3d function to use CPU instead of CUDA for mesh processing, ensuring compatibility with non-GPU environments. This change improves flexibility in data handling and maintains the integrity of the mesh generation process.
Update device handling in TwoStagePipeline to default to 'cuda' and adjust model loading to use 'cpu' for compatibility. Ensure consistent device usage across stages and improve code clarity.
Enhance Mesh class documentation by adding missing line breaks in docstrings for improved readability. Update device handling in FlexiCubes and FlexiCubesGeometry classes to default to 'cuda', ensuring consistent device usage across the application. Refactor ImageDreamDiffusion class to assert mode validity and streamline camera matrix pre-computation.
Refactor generate3d function in inference.py to enhance 3D mesh generation. Introduced new image processing steps for color and XYZ data, improved triplane feature extraction, and updated mesh export logic to utilize temporary file handling. This streamlines the rendering process and maintains compatibility with GPU devices.
Enhance generate3d function to initialize nvdiffrast context for mesh export, improving rendering capabilities. Updated export call to include rendering context and adjusted parameters for UVs and texture image, ensuring better integration with the model.