AttentionPainter: An Efficient and Adaptive Stroke Predictor for Scene Painting

Fuente: arXiv
Saved in:
Bibliographic Details
Main Authors: Tang, Yizhe, Wang, Yue, Hu, Teng, Yi, Ran, Tan, Xin, Ma, Lizhuang, Lai, Yu-Kun, Rosin, Paul L.
Format: Preprint
Published: 2024
Subjects:
Online Access:
Tags: Add Tag
No Tags, Be the first to tag this record!
_version_ 1866916453584732160
author Tang, Yizhe
Wang, Yue
Hu, Teng
Yi, Ran
Tan, Xin
Ma, Lizhuang
Lai, Yu-Kun
Rosin, Paul L.
author_facet Tang, Yizhe
Wang, Yue
Hu, Teng
Yi, Ran
Tan, Xin
Ma, Lizhuang
Lai, Yu-Kun
Rosin, Paul L.
contents Stroke-based Rendering (SBR) aims to decompose an input image into a sequence of parameterized strokes, which can be rendered into a painting that resembles the input image. Recently, Neural Painting methods that utilize deep learning and reinforcement learning models to predict the stroke sequences have been developed, but suffer from longer inference time or unstable training. To address these issues, we propose AttentionPainter, an efficient and adaptive model for single-step neural painting. First, we propose a novel scalable stroke predictor, which predicts a large number of stroke parameters within a single forward process, instead of the iterative prediction of previous Reinforcement Learning or auto-regressive methods, which makes AttentionPainter faster than previous neural painting methods. To further increase the training efficiency, we propose a Fast Stroke Stacking algorithm, which brings 13 times acceleration for training. Moreover, we propose Stroke-density Loss, which encourages the model to use small strokes for detailed information, to help improve the reconstruction quality. Finally, we propose a new stroke diffusion model for both conditional and unconditional stroke-based generation, which denoises in the stroke parameter space and facilitates stroke-based inpainting and editing applications helpful for human artists design. Extensive experiments show that AttentionPainter outperforms the state-of-the-art neural painting methods.
format Preprint
id arxiv_https___arxiv_org_abs_2410_16418
institution arXiv
publishDate 2024
record_format arxiv
spellingShingle AttentionPainter: An Efficient and Adaptive Stroke Predictor for Scene Painting
Tang, Yizhe
Wang, Yue
Hu, Teng
Yi, Ran
Tan, Xin
Ma, Lizhuang
Lai, Yu-Kun
Rosin, Paul L.
Computer Vision and Pattern Recognition
Stroke-based Rendering (SBR) aims to decompose an input image into a sequence of parameterized strokes, which can be rendered into a painting that resembles the input image. Recently, Neural Painting methods that utilize deep learning and reinforcement learning models to predict the stroke sequences have been developed, but suffer from longer inference time or unstable training. To address these issues, we propose AttentionPainter, an efficient and adaptive model for single-step neural painting. First, we propose a novel scalable stroke predictor, which predicts a large number of stroke parameters within a single forward process, instead of the iterative prediction of previous Reinforcement Learning or auto-regressive methods, which makes AttentionPainter faster than previous neural painting methods. To further increase the training efficiency, we propose a Fast Stroke Stacking algorithm, which brings 13 times acceleration for training. Moreover, we propose Stroke-density Loss, which encourages the model to use small strokes for detailed information, to help improve the reconstruction quality. Finally, we propose a new stroke diffusion model for both conditional and unconditional stroke-based generation, which denoises in the stroke parameter space and facilitates stroke-based inpainting and editing applications helpful for human artists design. Extensive experiments show that AttentionPainter outperforms the state-of-the-art neural painting methods.
title AttentionPainter: An Efficient and Adaptive Stroke Predictor for Scene Painting
topic Computer Vision and Pattern Recognition
url https://arxiv.org/abs/2410.16418