add overlay

zz44l43 · Jan 10, 2018 · 101a5be · 101a5be
1 parent 2e4c27c
commit 101a5be
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/train.py b/train.py
@@ -320,6 +320,9 @@ def perform_crf(image, probabilities):
                 crf_ed = perform_crf(val_orig_image, val_poss)
                 cv2.imwrite(os.path.join(FLAGS.output_dir, 'eval', 'val_{0}_prediction_crfed.jpg'.format(gs)), cv2.cvtColor(grayscale_to_voc_impl(np.squeeze(crf_ed)), cv2.COLOR_RGB2BGR))
 
+                overlay = cv2.addWeighted(cv2.cvtColor(np.squeeze(val_orig_image), cv2.COLOR_RGB2BGR), 1, cv2.cvtColor(grayscale_to_voc_impl(np.squeeze(crf_ed)), cv2.COLOR_RGB2BGR), 0.8, 0)
+                cv2.imwrite(os.path.join(FLAGS.output_dir, 'eval', 'val_{0}_overlay.jpg'.format(gs)), overlay)
+
     coord.request_stop()
     coord.join(threads)