From 9820af97d4cf34cdc06f99809411ed71ec69183f Mon Sep 17 00:00:00 2001 From: Russ Taylor Date: Tue, 10 Dec 2019 08:07:39 -0500 Subject: [PATCH] Adding parameter to allow pickle, which makes the code work with versions of numpy from 1.17.1. --- .../Computer Vision Toolkit/lib/Algorithms/AODNet.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/Computer Vision Toolkit/Computer Vision Toolkit/lib/Algorithms/AODNet.py b/Computer Vision Toolkit/Computer Vision Toolkit/lib/Algorithms/AODNet.py index 209daee..27241ab 100644 --- a/Computer Vision Toolkit/Computer Vision Toolkit/lib/Algorithms/AODNet.py +++ b/Computer Vision Toolkit/Computer Vision Toolkit/lib/Algorithms/AODNet.py @@ -92,7 +92,7 @@ def Dehaze(img_path, Params = None, model_path = 'lib/Algorithms/pretrained_aod_ x = np.expand_dims(img, axis=0).transpose(0,3,1,2) # Loading model weights - np_weights = np.load(model_path).item() + np_weights = np.load(model_path, allow_pickle=True).item() output = aod_net(x, np_weights) output = np.squeeze(output)