xjqjlu / CBStyling

Class-Based Styling

Real-time Localized Style Transfer with Semantic Segmentation

ICCV 2019 Computer Vision for Fashion, Art and Design

[Paper][Video]

Running the code

Given an input image, our method CBS can stylize a certain object class with a specif style. Run the following command to achieve the result below.

python main.py -i images/example.jpg -o images/stylized.jpg -s styles/mosaic.pth -c pedestrian

Input Image	Stylized Cars with CBS

Choosing the style and the object class

Styles	Object Classes

Citation

If you find the code useful for your research, please cite: