Views
No views yet
pip install git+https://github.com/msu-video-group/tsrn-rtvd1import numpy as np
2from tsrn_rtvd import TSRNRTVD
3
4model = TSRNRTVD.from_pretrained("egorchistov/deblurring-tsrn-rtvd", device="cuda")
5
6first = np.zeros((720, 1280, 3), dtype=np.uint8)
7next_frame = np.zeros((720, 1280, 3), dtype=np.uint8)
8
9model.reset(first)
10result = model.process_next(next_frame)
11deblurred_bgr = result.output_bgr