summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorCameron <cysmith1010@gmail.com>2016-11-07 22:07:28 -0700
committerGitHub <noreply@github.com>2016-11-07 22:07:28 -0700
commit5437aedcf027ab4f48a43b24b069efb4639db286 (patch)
tree776af50fb5242bfbc67b3b26ded1dd7fbae39c19
parent6432011ff3605f6c4aa6fbe02114b954c7f4db78 (diff)
Update neural_style.py
-rw-r--r--neural_style.py4
1 files changed, 1 insertions, 3 deletions
diff --git a/neural_style.py b/neural_style.py
index e199678..936abce 100644
--- a/neural_style.py
+++ b/neural_style.py
@@ -792,8 +792,6 @@ def get_content_weights(frame, prev_frame):
backward_path = os.path.join(args.video_input_dir, backward_fn)
forward_weights = read_weights_file(forward_path)
backward_weights = read_weights_file(backward_path)
- forward_weights = np.clip(forward_weights, 0, 255).astype('uint8')
- backward_weights = np.clip(backward_weights, 0, 255).astype('uint8')
return forward_weights #, backward_weights
def warp_image(src, flow):
@@ -874,4 +872,4 @@ def main():
else: render_single_image()
if __name__ == '__main__':
- main() \ No newline at end of file
+ main()