search

wonjongg / StyleCariGAN

[SIGGRAPH 2021] Official implementation of "StyleCariGAN: Caricature Generation via StyleGAN Feature Map Modulation"
GNU Affero General Public License v3.0
295 stars 53 forks source link

Attribute classifiers #17

Closed WangJYao closed 1 year ago

WangJYao commented 1 year ago

How to train the attribute classifiers? Where is the training code？ Thank you.

wonjongg commented 1 year ago

We trained the attribute classifier with WebCariA dataset by following their protocols. The network architecture & loss function for training are ResNet-18 & binary cross-entropy loss in PyTorch (please see page 9 in our paper) The training code for the attribute classifier is not included in this project.

Thank you!