Neural style transfer pipeline for arbitrary stylization with strength control and color transfer control.
Paper on Russian (diploma work): "Modification of automatic stylization method via transformer network"
Code for style transfer module based on this repository.
Model weights (checkpoint): link
Checkpoint has the following keys: "model_state_dict", "optimizer_state_dict", "epoch", "loss". So, to get model's weights, you should use only "model_state_dict".