diff options
| -rw-r--r-- | neural_style.py | 8 |
1 files changed, 4 insertions, 4 deletions
diff --git a/neural_style.py b/neural_style.py index b717216..d5c5eaa 100644 --- a/neural_style.py +++ b/neural_style.py @@ -158,11 +158,11 @@ def parse_args(): help='Boolean flag indicating if the user is generating a video.') parser.add_argument('--start_frame', type=int, - default=1, + default=1, help='First frame number.') parser.add_argument('--end_frame', type=int, - default=1, + default=1, help='Last frame number.') parser.add_argument('--first_frame_type', type=str, @@ -525,8 +525,8 @@ def read_flow_file(path): flow = np.ndarray((2, h, w), dtype=np.float32) for y in range(h): for x in range(w): - flow[1,y,x] = struct.unpack('f', f.read(4))[0] flow[0,y,x] = struct.unpack('f', f.read(4))[0] + flow[1,y,x] = struct.unpack('f', f.read(4))[0] return flow def read_weights_file(path): @@ -869,4 +869,4 @@ def main(): else: render_single_image() if __name__ == '__main__': - main() +main()
\ No newline at end of file |