InvokeAI/ldm
Lincoln Stein f816526d0d add ability to post-process images from the CLI
- supports gfpgan, esrgan, codeformer and embiggen
- To use:
   dream> !fix ./outputs/img-samples/000056.292144555.png -ft gfpgan -U2 -G0.8
   dream> !fix ./outputs/img-samples/000056.292144555.png -ft codeformer -G 0.8
   dream> !fix ./outputs/img-samples/000056.29214455.png  -U4
   dream> !fix ./outputs/img-samples/000056.292144555.png -embiggen 1.5

   The first example invokes gfpgan to fix faces and esrgan to upscale.
   The second example invokes codeformer to fix faces, no upscaling
   The third example uses esrgan to upscale 4X
   The four example runs embiggen to enlarge 1.5X

- This is very preliminary work. There are some anomalies to note:
  1. The syntax is non-obvious. I would prefer something like:
     !fix esrgan,gfpgan
     !fix esrgan
     !fix embiggen,codeformer

     However, this will require refactoring the gfpgan and embiggen
     code.

   2. Images generated using gfpgan, esrgan or codeformer all are named
      "xxxxxx.xxxxxx.postprocessed.png" and the original is saved.
      However, the prefix is a new one that is not related to the
      original.

   3. Images generated using embiggen are named "xxxxx.xxxxxxx.png",
      and once again the prefix is new. I'm not sure whether the
      prefix should be aligned with the original file's prefix or not.
      Probably not, but opinions welcome.
2022-09-19 12:38:14 -04:00
..
data prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
dream add ability to post-process images from the CLI 2022-09-19 12:38:14 -04:00
models Refactor generate.py and dream.py (#534) 2022-09-14 07:02:31 -04:00
modules Enable even larger images with one simple torch.nn.functional.silu import (#653) 2022-09-17 18:03:52 -04:00
restoration GFPGAN and Real ESRGAN Implementation Refactor 2022-09-19 23:38:56 +12:00
generate.py add ability to post-process images from the CLI 2022-09-19 12:38:14 -04:00
lr_scheduler.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00
simplet2i.py Disabled debug output (#436) 2022-09-11 22:34:35 -04:00
util.py prettified all the code using "blue" at the urging of @tildebyte 2022-08-26 03:15:42 -04:00