Switch net to eval mode in predict.py

Former-commit-id: 3af421f3060c30b19ee38bc6e00db2593e53ad62
2018-09-26 08:59:49 +02:00 · 2018-09-26 08:59:49 +02:00 · f2d88f98a7
parent 341824fb06
commit f2d88f98a7
1 changed files with 1 additions and 0 deletions
--- 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]