Skip to content

Commit

Permalink
Update kitti_seg_input.py (#81)
Browse files Browse the repository at this point in the history
- scipy.misc.imread reads HWC, not WHC; in all other parts of the code outside this docstring, height is in the 0th dim.
  • Loading branch information
villanuevab authored and MarvinTeichmann committed Aug 8, 2017
1 parent 860c781 commit 234bc43
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions inputs/kitti_seg_input.py
Original file line number Diff line number Diff line change
Expand Up @@ -129,9 +129,9 @@ def _make_data_gen(hypes, phase, data_dir):
"""Return a data generator that outputs image samples.
@ Returns
image: integer array of shape [width, height, 3].
image: integer array of shape [height, width, 3].
Representing RGB value of each pixel.
gt_image: boolean array of shape [width, height, num_classes].
gt_image: boolean array of shape [height, width, num_classes].
Set `gt_image[i,j,k] == 1` if and only if pixel i,j
is assigned class k. `gt_image[i,j,k] == 0` otherwise.
Expand Down

0 comments on commit 234bc43

Please sign in to comment.