gvecchio/MatForger · Hugging Face

Three Textures for the Designers under the sky,
Seven for the Artists in their studios of light,
Nine for the Architects doomed to try,
One for the Developers on their screens so bright
In the Land of Graphics where the Pixels lie.
One Forge to craft them all, One Code to find them,
One Model to bring them all and to the mesh bind them
In the Land of Graphics where the Pixels lie.

^{_{Our deep apologies to J. R. R. Tolkien}}

🤖 Model Details

Overview

MatForger is a generative diffusion model designed specifically for generating Physically Based Rendering (PBR) materials. Inspired by the MatFuse model and trained on the comprehensive MatSynth dataset, MatForger pushes the boundaries of material synthesis. It employs the noise rolling technique, derived from ControlMat, to produce tileable maps. The model generates multiple maps, including basecolor, normal, height, roughness, and metallic, catering to a wide range of material design needs.

Features

High-Quality PBR Material Generation: Produces detailed and realistic materials suited for various applications.
Tileable Textures: Utilizes a noise rolling approach to ensure textures are tileable, enhancing their usability in larger scenes.
Versatile Outputs: Generates multiple texture maps (basecolor, normal, height, roughness, metallic) to meet the requirements of complex material designs.
Text and Image Conditioning: Can be conditioned with either images or text inputs to guide material generation, offering flexibility in creative workflows.

Model Description

MatForger architecture is based on MatFuse. It differs from it in using a continuous VAE instead of a vector quantized autoencoder (VQ-VAE). Additionally we distilled the multiencoder VAE into a single-encoder model, thus reducing the model complexity but retaining the disentangled latent representation of MatFuse.

⚒️ MatForger at work

MatForger can be conditioned via text prompts or images to generate high-quality materials. Following some examples of materials generated using MatForge. For each sample we report the prompt, the generated maps (basecolor, normal, height, roughness, metallic) and the resulting rendering.

Text2Mat samples

Image2Mat samples

🧑‍💻 How to use

MatForger requires a custom pipeline due to the data type.

You can use it in 🧨 diffusers:

import torch

from PIL import Image

from diffusers import DiffusionPipeline

device = torch.device("cuda" if torch.cuda.is_available() else "cpu")

pipe = DiffusionPipeline.from_pretrained(
    "gvecchio/MatForger",
    trust_remote_code=True,
)

pipe.enable_vae_tiling()

pipe.enable_freeu(s1=0.9, s2=0.2, b1=1.1, b2=1.2)
pipe.to(device)

# model prompting with image
prompt = Image.open("bricks.png")
image = pipe(
    prompt,
    guidance_scale=6.0,
    height=512,
    width=512,
    tileable=True, # Allows to generate tileable materials
    patched=False, # Reduce memory requirements for high-hes generation but affects quality 
    num_inference_steps=25,
).images[0]

# model promptiong with text
prompt = "terracotta brick wall"
image = pipe(
    prompt,
    guidance_scale=6.0,
    height=512,
    width=512,
    tileable=True, # Allows to generate tileable materials
    patched=False, # Reduce memory requirements for high-hes generation but affects quality 
    num_inference_steps=25,
).images[0]

# get maps from prediction
basecolor = image.basecolor
normal = image.normal
height = image.height
roughness = image.roughness
metallic = image.metallic

⚠️ Bias and Limitations

Quality: The model was trained on a variety of synthetic and real data from the MatSynth dataset. However, it might fail to generate complex materials or patterns that differ significantly from the training data distribution. If the results don't meet your expectations be patient and keep trying.

Resolution: MatForger can generate high resolution materials via latent patching. The model, however, was trained at a 256x256 resolution and artifacts or inconsistencies might appear at higher resolution.

❗Note: MatForge is a home-trained model, with limited resources. We will try to update it regularly and improve its performances.
We welcome contributions, feedback, and suggestions to enhance its capabilities and address its limitations. Please be patient as we work towards making MatForger an even more powerful tool.

💡 Upcoming features ideas

As MatForger continues to evolve, we're working on several features aimed at enhancing its utility and effectiveness. As we continue to refine and expand its capabilities, here are some of the possible upcoming enhancements:

Opacity: Generate opacity map for materials requiring transparency.
Material Inpainting: A feature designed to allow users to modify and enhance materials by filling in gaps or correcting imperfections directly within the generated textures.
Sketch-Based Material Generation: We're exploring ways to convert simple sketches into detailed materials. This aims to simplify the material creation process, making it more accessible to users without in-depth technical expertise.
Color Palette Conditioning: Future updates will offer improved control over the color palette of generated materials, enabling users to achieve more precise color matching for their projects.
Material Estimation from Photographs: We aim to refine the model's ability to interpret and recreate the material properties observed in photographs, facilitating the creation of materials that closely mimic real-world textures.

🎯 Ongoing Development and Openness to Feedback

MatForger is a RESEARCH TOOL, thus its development is an ongoing process and highly subsceptible to our research agenda. However, we are committed to improving MatForger's capabilities and addressing any limitations and implementing suggestions we receive from our users.

🤝 How to Contribute

Feature Suggestions: If you have ideas for new features or improvements, we're eager to hear them. Reach out to us! Your suggestions play a crucial role in guiding the direction of MatForger's development.

Dataset Contributions: Enhancing the diversity of our training data can significantly improve the model's performance. If you have access to textures, materials, or data that could benefit MatForger, consider contributing.

Feedback: User feedback is invaluable for identifying areas for improvement. Whether it's through reporting issues or sharing your experiences, your insights help us make MatForger better.

Training: If you have spare compute resources and wish to contribute to the training of MatForger reach out to [email protected].

Terms of Use:

We hope that the release of this model will make community-based research efforts more accessible. This model is governed by a Openrail License and is intended for research purposes.

gvecchio
/

MatForger