You can deploy this model and start earning money today!

Druudik1/
watercolor-brush

Name: watercolor-brush
Availability: InStock

mit

Views

No views yet

This repository contains trained ColorMaskNeuralRenderer that learned to mimic WatercolorBrush (these are components of my Stylized Neural Painting paper reimplementation).

Use the following code snippet to load it and create the corresponding WatercolorBrush:

python

from huggingface_hub import hf_hub_download
from painting.brushes import WatercolorBrush, ColorMaskNeuralRenderer

# Specify your device and output canvas size.
device = "cuda"
out_canvas_size = 512

brush = WatercolorBrush(canvas_size=out_canvas_size, device=device)
differentiable_brush = ColorMaskNeuralRenderer(
  brush_params_count=brush.brush_params_count,
  color_params_indices=[7, 8, 9],
  canvas_size=128,
  device=device,
)

weights = hf_hub_download(repo_id="Druudik1/watercolor-brush", filename="neural_renderer.pt")
differentiable_brush.load_state_dict(
    torch.load(weights, map_location=device, weights_only=True)
)