Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Colab Questions #7

Open
noobtoob4lyfe opened this issue Sep 15, 2021 · 4 comments
Open

Colab Questions #7

noobtoob4lyfe opened this issue Sep 15, 2021 · 4 comments

Comments

@noobtoob4lyfe
Copy link

Thank you so much for sharing the this!
Forgive me, because I'm not super technical but I was hoping you could walk me through how to use the Colab notebook to test my own image sequence. When I "run all" in Colab it shows a single output comparison jpeg. Does it save the entire deblurred image sequence anywhere? All I can see are the input images when I search the folder structure. How would I deblur and entire image sequence and save the resulting images? Thanks in advance!

@ljzycmd
Copy link
Owner

ljzycmd commented Sep 17, 2021

If you want to output all frames, you should add a FOR-LOOP in Sec.4 of the Colab Notebook. Just like the following:

for i, batch_data in enumerate(frames_data): 
    out = model(batch_data["input_frames"].unsqueeze(0).cuda())
    out = out.clamp(0, 1)
    # save the output
    ...

You also need to implement saving the output frames.

@noobtoob4lyfe
Copy link
Author

Awesome! Thanks so much for your help. So is that what the full text of Sec.4 should look like, or would that snippet be pasted in after a certain line in the existing text. How would one implement the "saving the output frames" aspect of this? Also what would I change if I want to try the ESTRNN model instead of the DBN model? Thanks again!

@robbsaber
Copy link

I would also like to know how to save the output frames.

@ljzycmd
Copy link
Owner

ljzycmd commented Feb 20, 2022

Yep, I will update the notebook with saving the output frames soon!

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

3 participants