diff --git a/image_dehazer/__init__.py b/image_dehazer/__init__.py index 9c54b81..f4a414e 100644 --- a/image_dehazer/__init__.py +++ b/image_dehazer/__init__.py @@ -257,7 +257,7 @@ def __zero_pad(self, image, shape, position='corner'): shape = np.asarray(shape, dtype=int) imshape = np.asarray(image.shape, dtype=int) - if np.alltrue(imshape == shape): + if np.all(imshape == shape): return image if np.any(shape <= 0): @@ -299,4 +299,4 @@ def remove_haze(HazeImg, airlightEstimation_windowSze=15, boundaryConstraint_win regularize_lambda=regularize_lambda, sigma=sigma, delta=delta, showHazeTransmissionMap=showHazeTransmissionMap) HazeCorrectedImg, HazeTransmissionMap = Dehazer.remove_haze(HazeImg) - return (HazeCorrectedImg, HazeTransmissionMap) \ No newline at end of file + return (HazeCorrectedImg, HazeTransmissionMap)