Skip to content
This repository was archived by the owner on Jan 7, 2025. It is now read-only.

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
Contributor

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 subscribe to this conversation on GitHub. Already have an account? Sign in.

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants