Downloads · 30 days
0
Druudik1/rectangle-brush
rectangle-brush is a machine learning model from Druudik1. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Downloads · 30 days
0
Access
Public
Updated Feb 23, 2026
Repo size
44.2 MB
Likes
0
Public
Click a slice to open those files.
.pt44.2 MB · 100%
From the Hugging Face model README
This repository contains trained GeneralNeuralRenderer that learned to mimic RectangleBrush (these are
components of my Stylized Neural Painting paper reimplementation).
Use the following code snippet to load it and create the corresponding RectangleBrush:
from huggingface_hub import hf_hub_download
from painting.brushes import RectangleBrush, GeneralNeuralRenderer
# Specify your device and output canvas size.
device = "cuda"
out_canvas_size = 512
brush = RectangleBrush(canvas_size=out_canvas_size, device=device)
differentiable_brush = GeneralNeuralRenderer(
brush_params_count=brush.brush_params_count,
canvas_size=128,
device=device,
)
weights = hf_hub_download(repo_id="Druudik1/rectangle-brush", filename="neural_renderer.pt")
differentiable_brush.load_state_dict(
torch.load(weights, map_location=device, weights_only=True)
)