Unable to execute inference_image script

[Aparajit-Garg]

Hi,
Great repository, just loved it. I am trying to execute inference_image.py with the dbn architecture as the backbone but I'm getting stuck on this:
Executing the script produces the following:

Cannot inport EDVR modules!!!
Cannot import STFAN modules!!!
Using checkpoint loaded from ./checkpoints/dbn_ckpt.pth for testing.
Traceback (most recent call last):
File "inference_image.py", line 81, in
inference()
File "inference_image.py", line 70, in inference
outputs = arch.postprocess(arch.model(arch.preprocess(input_image)))
File "/usr/local/lib/python3.7/dist-packages/torch/nn/modules/module.py", line 1102, in _call_impl
return forward_call(*input, **kwargs)
File "/content/drive/MyDrive/SimDeblur/simdeblur/model/backbone/dbn/dbn.py", line 137, in forward
central_frame = x[:, self.num_frames // 2]
IndexError: index 2 is out of bounds for dimension 1 with size 1

I tried printing the dimensions of the image and it came out to be: torch.Size([1, 1, 3, 385, 1504])
and self.num_frames is 5 for my case.
Don't know how to resolve the issue. Please help out.

[ljzycmd]

Hi, note that EDVR and STFAN models are required to be compiled on the GPU platform, and the script "inference_image.py" is just responsible for single image deblurring models, e.g., SRN. If you want to infer a video, you can try the notebook on CoLab. Meanwhile, I will release "inference_video.py" for video deblurring models soon, please stay tuned!