From 939faca825404d24556742b26b202c293964bae7 Mon Sep 17 00:00:00 2001 From: Jonathan Böcker Date: Sun, 4 Dec 2016 17:13:54 +0100 Subject: Python 3 compatibility Map objects is not subscriptable in Python 3 --- neural_style.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'neural_style.py') diff --git a/neural_style.py b/neural_style.py index 5373e5e..8d40e1b 100644 --- a/neural_style.py +++ b/neural_style.py @@ -535,14 +535,14 @@ def read_flow_file(path): def read_weights_file(path): lines = open(path).readlines() - header = map(int, lines[0].split(' ')) + header = list(map(int, lines[0].split(' '))) w = header[0] h = header[1] vals = np.zeros((h, w), dtype=np.float32) for i in range(1, len(lines)): line = lines[i].rstrip().split(' ') - vals[i-1] = np.array(map(np.float32, line)) - vals[i-1] = map(lambda x: 0. if x < 255. else 1., vals[i-1]) + vals[i-1] = np.array(list(map(np.float32, line))) + vals[i-1] = list(map(lambda x: 0. if x < 255. else 1., vals[i-1])) # expand to 3 channels weights = np.dstack([vals.astype(np.float32)] * 3) return weights -- cgit v1.2.3-70-g09d2