forked from StacyYang/MSG-Net
-
Notifications
You must be signed in to change notification settings - Fork 0
/
init.lua
31 lines (28 loc) · 1.01 KB
/
init.lua
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
--+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
-- Created by: Hang Zhang
-- ECE Department, Rutgers University
-- Email: zhang.hang@rutgers.edu
-- Copyright (c) 2017
--
-- Free to reuse and distribute this software for research or
-- non-profit purpose, subject to the following conditions:
-- 1. The code must retain the above copyright notice, this list of
-- conditions.
-- 2. Original authors' names are not deleted.
-- 3. The authors' names are not used to endorse or promote products
-- derived from this software
--+++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++++
require 'nn'
require 'cunn'
require 'cudnn'
-- load packages from perceptual-loss
require 'texture.utils'
require 'texture.GramMatrix'
require 'texture.ContentLoss'
require 'texture.StyleLoss'
require 'texture.TotalVariation'
require 'texture.PerceptualCriterion'
require 'texture.InstanceNormalization'
-- load MSG-Net dependencies
require 'texture.Calibrate'
require 'texture.Inspiration'