Skip to content

Instantly share code, notes, and snippets.

@ProGamerGov
ProGamerGov / neural_style_seg.lua
Last active December 27, 2018 03:02
Working Multi-Region Spatial Control in Neural-Style. Also known as "masked style transfer", and "semantic/segmented style transfer".
-- Original mask related code from: https://github.com/martinbenson/deep-photo-styletransfer
-- Modified mask code by github.com/ProGamerGov
require 'torch'
require 'nn'
require 'image'
require 'optim'
require 'loadcaffe'