Skip to content

Conversation

psychedelicious
Copy link
Collaborator

I've opted to leave out any additional upscaling parameters like scale and denoising strength, which, from my review of the ESRGAN code, don't do much:

  • scale just resizes the image using CV2 after the AI upscaling, so that's not particularly useful
  • denoising strength is only valid for one class of model, which we are no longer supporting

If there is demand, we can implement output size/scale UI and handle it by passing the upscaled image to that a resize/scale node.

I also understand we previously had some functionality to blend the upscaled image with the original. If that is desired, we would need to implement that as a node that we can pass the upscaled image to.

Demo:

Screen.Recording.2023-07-17.at.9.22.01.pm.mov

feat(nodes): add RealESRGAN_x2plus.pth, update upscale nodes

  • add RealESRGAN_x2plus.pth model to installer @lstein
  • add RealESRGAN_x2plus.pth to realesrgan node
  • rename RealESRGAN to ESRGAN in nodes
  • make scale_factor optional in img_scale node

feat(ui): restore ad-hoc upscaling

  • remove face restoration entirely
  • add dropdown for ESRGAN model select
  • add ad-hoc upscaling graph and workflow

@psychedelicious
Copy link
Collaborator Author

I've renamed the node esrgan but maybe it's better named realesrgan? I dunno.

Copy link
Collaborator

@lstein lstein left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Code looks good and works well.

- add `RealESRGAN_x2plus.pth` model to installer
- add `RealESRGAN_x2plus.pth` to `realesrgan` node
- rename `RealESRGAN` to `ESRGAN` in nodes
- make `scale_factor` optional in `img_scale` node
- remove face restoration entirely
- add dropdown for ESRGAN model select
- add ad-hoc upscaling graph and workflow
@psychedelicious
Copy link
Collaborator Author

Fixed tests, forgot to change something when I renamed a node

@lstein lstein merged commit c89fa4b into main Jul 19, 2023
@lstein lstein deleted the feat/ui/upscale branch July 19, 2023 01:00
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.

2 participants