Skip to content

Commit

Permalink
modified test model script
Browse files Browse the repository at this point in the history
  • Loading branch information
MaybeShewill-CV committed May 29, 2018
1 parent a9650d2 commit c47d231
Showing 1 changed file with 0 additions and 2 deletions.
2 changes: 0 additions & 2 deletions tools/test_lanenet.py
Original file line number Diff line number Diff line change
Expand Up @@ -118,7 +118,6 @@ def test_lanenet(image_path, weights_path, use_gpu):
# mask_image = cv2.resize(mask_image, (image_vis.shape[1], image_vis.shape[0]),
# interpolation=cv2.INTER_LINEAR)

# plt.ion()
plt.figure('mask_image')
plt.imshow(mask_image[:, :, (2, 1, 0)])
plt.figure('src_image')
Expand All @@ -128,7 +127,6 @@ def test_lanenet(image_path, weights_path, use_gpu):
plt.figure('binary_image')
plt.imshow(binary_seg_image[0] * 255, cmap='gray')
plt.show()
# plt.pause(3.0)

sess.close()

Expand Down

0 comments on commit c47d231

Please sign in to comment.