Skip to content

Commit bcd22a9

Browse files
authored
Update image_reader.py
1 parent a8f0eba commit bcd22a9

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

utils/image_reader.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -100,6 +100,7 @@ def _random_crop_and_pad_image_and_labels(image, label, crop_h, crop_w, ignore_l
100100
combined_crop = tf.random_crop(combined_pad, [crop_h, crop_w, 4])
101101
img_crop = combined_crop[:, :, :last_image_dim]
102102
label_crop = combined_crop[:, :, last_image_dim:]
103+
label_crop = label_crop + ignore_label
103104
label_crop = tf.cast(label_crop, dtype=tf.uint8)
104105

105106
# Set static shape so that tensorflow knows shape at compile time.

0 commit comments

Comments
 (0)