From f2d88f98a72ff4a0e692ee37e28374df7f3b3cbb Mon Sep 17 00:00:00 2001
From: milesial <milesial@users.noreply.github.com>
Date: Wed, 26 Sep 2018 08:59:49 +0200
Subject: [PATCH]  Switch net to eval mode in predict.py

Former-commit-id: 3af421f3060c30b19ee38bc6e00db2593e53ad62
---
 predict.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/predict.py b/predict.py
index bb48e46..5347d88 100755
--- a/predict.py
+++ b/predict.py
@@ -20,6 +20,7 @@ def predict_img(net,
                 use_dense_crf=True,
                 use_gpu=False):
 
+    net.eval()
     img_height = full_img.size[1]
     img_width = full_img.size[0]