Skip to content

Commit

Permalink
minor cleanup
Browse files Browse the repository at this point in the history
  • Loading branch information
Ryan Zotti authored and Ryan Zotti committed Aug 28, 2016
1 parent 3ae8c42 commit 30e2725
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions play_numpy_dataset.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
import cv2
import numpy as np

input_file_path = '/Users/ryanzotti/Documents/repos/Self_Driving_RC_Car/data/final_processed_data.npz'
input_file_path = '/Users/ryanzotti/Documents/repos/Self_Driving_RC_Car/final_processed_data_3_channels.npz'
npzfile = np.load(input_file_path)

train_predictors = npzfile['train_predictors']
Expand All @@ -10,10 +10,9 @@
print(train_predictors.shape)

for frame_index, frame in enumerate(train_predictors):
print(frame_index)

cv2.imshow('frame', frame)

# This line is necessary or else it will look like the frames are frozen
if cv2.waitKey(1) & 0xFF == ord('q'):
break
print()

0 comments on commit 30e2725

Please sign in to comment.