PyDMD / PyDMD

Python Dynamic Mode Decomposition

Home Page:https://pydmd.github.io/PyDMD/

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

12th tutorial - compressed DMD - function doesn't return noisy video after adding noise

karlic-luka opened this issue · comments

Describe the bug
In the 12th tutorial - compressed DMD - there's an option to add the noise to the original video.
However, the function returns a video without noise (even if you added it).

Original method:

def get_video_dmd(
    object: str = "frog", noise: bool = False, noise_amt: float = 0.01
) -> Tuple[np.ndarray, Tuple[int, int]]:
...
 vid = np.vstack(imgs).T
    if noise:
        vid += np.random.normal(0, noise_amt, vid.shape)
        vid = vid.clip(0, 1)
    return np.vstack(imgs).T, shape

Return statement gives np.vstack(imgs).T, while it should be only vid