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

Enable the ability to save the RRDN model #137

Merged
merged 5 commits into from Sep 8, 2020

Conversation

thekevinscott
Copy link
Contributor

@thekevinscott thekevinscott commented Jul 31, 2020

Pursuant to this comment thread, enables the ability to save the RRDN model.

I tried to implement a get_config but was unsuccessful, so I copied the implementation by @gkrislara. I'm using this code to save an RRDN model successfully.

Sample code for saving:

from ISR.models import RRDN
model = RRDN(weights='gans')
model.model.save('./image-super-resolution-gans.h5')

@thekevinscott
Copy link
Contributor Author

I've updated the PR to use custom layers; these also leverage get_config as referenced above.

@uiyunkim-git
Copy link

uiyunkim-git commented Sep 8, 2020

Just adopted your rrdn.py

The error "can't pickle _thread.RLock objects" has fixed.

Thanks!

@cfrancesco cfrancesco merged commit b821af9 into idealo:master Sep 8, 2020
@cfrancesco
Copy link
Collaborator

looks great, thanks!

thekevinscott pushed a commit to thekevinscott/image-super-resolution that referenced this pull request Oct 1, 2022
Enable the ability to save the RRDN model
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

Successfully merging this pull request may close these issues.

None yet

3 participants