diff --git a/README.md b/README.md index 72d87b8..5744a37 100644 --- a/README.md +++ b/README.md @@ -24,6 +24,11 @@ does not attempt to recreate all of the matlab code from matlabPyrTools. The goal is to create a Python interface for the C code at the heart of matlabPyrTools. +**NOTE**: If you are only interested in the complex steerable pyramid, we have a +pytorch implementation in the +[plenoptic](https://github.com/LabForComputationalVision/plenoptic/) package; +this implementation is differentiable. + # Installation It's recommended you install from pip: `pip install pyrtools`. The pip diff --git a/pyrtools/pyramids/pyramid.py b/pyrtools/pyramids/pyramid.py index 54b9b8b..9670582 100644 --- a/pyrtools/pyramids/pyramid.py +++ b/pyrtools/pyramids/pyramid.py @@ -49,7 +49,7 @@ class Pyramid: def __init__(self, image, edge_type): - self.image = np.array(image).astype(np.float) + self.image = np.array(image).astype(float) if self.image.ndim == 1: self.image = self.image.reshape(-1, 1) assert self.image.ndim == 2, "Error: Input signal must be 1D or 2D." diff --git a/pyrtools/tools/display.py b/pyrtools/tools/display.py index 2a07856..4f03098 100644 --- a/pyrtools/tools/display.py +++ b/pyrtools/tools/display.py @@ -815,7 +815,7 @@ def animshow(video, framerate=2., as_html5=True, repeat=False, def animate_video(t): for i, a in enumerate(artists): - frame = video[i][t].astype(np.float) + frame = video[i][t].astype(float) a.set_data(frame) return artists