1
0
Fork 0
Commit Graph

243 Commits

Author SHA1 Message Date
Sean Sube 4d0898a52c feat(api): start adding model sources to convert script 2023-01-16 17:08:59 -06:00
Sean Sube 5a01fe4cb0 fix(api): resolve face correction model relative to model path 2023-01-16 16:45:17 -06:00
Sean Sube 1d90f9b53c fix class method 2023-01-16 16:40:59 -06:00
Sean Sube 4809e00982 fix(api): get all server paths from environ 2023-01-16 16:40:32 -06:00
Sean Sube c98c0ff4dd lint(api): remove global paths entirely 2023-01-16 16:39:30 -06:00
Sean Sube 9f4383716e feat(api): add ESRGAN/GFPGAN deps 2023-01-16 15:22:31 -06:00
Sean Sube 4bf68759d7 fix(api): report accurate sizes 2023-01-16 15:11:40 -06:00
Sean Sube d406cd4e99 fix(api): pass model to ONNX instances 2023-01-16 14:58:27 -06:00
Sean Sube adeada19e5 fix exports 2023-01-16 14:54:45 -06:00
Sean Sube 8d3ebede5a feat: add outscaling option 2023-01-16 14:52:56 -06:00
Sean Sube 091c4e6109 fix(api): pass upscale params when creating RESRGAN 2023-01-16 14:17:50 -06:00
Sean Sube d1e4fa9cf1 feat: add upscale controls to client, params to server 2023-01-16 14:05:04 -06:00
Sean Sube 1f0c19af04 fix(api): get upscale params from request 2023-01-16 13:12:08 -06:00
Sean Sube 120056f878 fix(api): get ESRGAN/GFPGAN paths from server context, clean up test scripts 2023-01-16 13:02:15 -06:00
Sean Sube 0f2d6d2ec7 add subscript for fake image 2023-01-16 12:02:37 -06:00
Sean Sube 402fa4f687 return fake image throughout chain 2023-01-16 11:46:06 -06:00
Sean Sube f493246192 fix(api): correct stub method name 2023-01-16 11:45:31 -06:00
Sean Sube ba3eff5c03 fix(api): enable tiling when fixing faces after upscaling 2023-01-16 11:43:00 -06:00
Sean Sube b37f8cdd7d stub out other methods for output image 2023-01-16 11:16:06 -06:00
Sean Sube 30d474b487 wrap image output 2023-01-16 11:06:50 -06:00
Sean Sube fe657468bf fix(api): add missing params to load stub 2023-01-16 10:59:47 -06:00
Sean Sube 9519fc16e9 feat(api): add ONNX implementation of Real ESRGAN net 2023-01-16 10:55:40 -06:00
Sean Sube 48963fa591 feat(api): split up test scripts for diffusers and real esrgan 2023-01-16 10:55:24 -06:00
Sean Sube 5fded3cf40 remove unused param from output 2023-01-16 10:03:30 -06:00
Sean Sube e59449fec1 feat(api): add conversion script for models 2023-01-16 09:57:59 -06:00
Sean Sube 1283bc3d3f fix(api): look up noise coordinates correctly 2023-01-16 07:49:25 -06:00
Sean Sube 604cdf39b0 lint(api): helper to get borders from request 2023-01-16 07:45:50 -06:00
Sean Sube 53fcc0b91f lint(api): remove output path, join from context 2023-01-16 07:42:10 -06:00
Sean Sube c8944a2be8 fix path 2023-01-16 07:36:35 -06:00
Sean Sube 7c9dbf2c7e remove unused fields from context 2023-01-16 07:34:37 -06:00
Sean Sube 8a97cbebb6 remove invalid model path call 2023-01-16 07:33:53 -06:00
Sean Sube 7c52743aaa remove invalid imports 2023-01-16 07:33:22 -06:00
Sean Sube 13a4fa2278 lint(api): encapsulate paths in server context class 2023-01-16 07:31:42 -06:00
Sean Sube a76793d105 fix(api): correct output paths, read strength from params 2023-01-15 20:00:26 -06:00
Sean Sube f9c33486c1 fix latents for txt2img 2023-01-15 19:49:40 -06:00
Sean Sube 7d8ea23166 fix param hashing 2023-01-15 19:48:43 -06:00
Sean Sube bdc8e277fe lint(api): clean up output file hash stuff 2023-01-15 19:47:57 -06:00
Sean Sube e43238b327 fix init imports 2023-01-15 19:34:51 -06:00
Sean Sube 44c81d966f lint(api): finish splitting up modules 2023-01-15 19:33:40 -06:00
Sean Sube ff70d36a05 lint(api): add class for params, size, other common data 2023-01-15 19:14:58 -06:00
Sean Sube 1146118b1a lint(api): split up modules 2023-01-15 18:54:20 -06:00
Sean Sube c7c3645466 lint(api): split out pipeline code 2023-01-15 18:46:00 -06:00
Sean Sube 806503c709 fix(api): load upscaling model from models dir 2023-01-15 18:41:58 -06:00
Sean Sube 45d65d1342 fix(api): convert back to PIL after upscaling 2023-01-15 18:25:12 -06:00
Sean Sube 1fe6fa91fb fix(api): convert image to numpy before upscaling 2023-01-15 18:13:28 -06:00
Sean Sube 77cb84c60e feat(api): add basic upscaling 2023-01-15 18:04:10 -06:00
Sean Sube ae8b26fa8f chore: bump package versions 2023-01-15 15:13:47 -06:00
Sean Sube 5bb3f6c558 feat: add noise source with solid color 2023-01-15 14:26:04 -06:00
Sean Sube 16108ae172 fix(api): add time to filenames (for #55) 2023-01-15 14:22:51 -06:00
Sean Sube df6b07194b fix(api): set default mask filter to none, matching the client 2023-01-15 14:06:37 -06:00
Sean Sube 40080ad46b feat: add gaussian multiply to mask filters 2023-01-15 14:04:54 -06:00
Sean Sube eedea93ade fix(api): blend source and noise in correct order 2023-01-15 13:47:39 -06:00
Sean Sube f142418707 fix(api): convert mask before blending source 2023-01-15 13:45:37 -06:00
Sean Sube 081a96d536 fix(api): use full-image compositing, write debug images to output dir 2023-01-15 13:40:11 -06:00
Sean Sube 9c93e16698 fix(api): clean up background jobs once they are ready (for #55) 2023-01-15 11:43:47 -06:00
Sean Sube e4020cf3f6 fix(api): always apply mask filter for inpainting 2023-01-15 11:39:40 -06:00
Sean Sube b1aca928ab fix(api): correct type of provider in output path 2023-01-15 11:29:38 -06:00
Sean Sube e429bafeef fix(api): include all parameters in output path 2023-01-15 11:28:12 -06:00
Sean Sube f5ed77a349 fix(api): reduce copies, fix function signatures 2023-01-15 11:09:47 -06:00
Sean Sube ef06b45599 fix(api): reuse results of blur modes 2023-01-15 10:54:17 -06:00
Sean Sube e2d17e1833 fix(api): fill gaussian blur with noise first 2023-01-15 10:49:09 -06:00
Sean Sube 4af1b8e1aa save mask and noise for debugging 2023-01-15 10:09:57 -06:00
Sean Sube 6c07d124e0 feat(api): add a mask filter to blend outpainting edges 2023-01-15 09:54:55 -06:00
Sean Sube 4675f89bb7 fix(api): add missing origin argument to noise sources 2023-01-15 09:40:08 -06:00
Sean Sube 0ed11af34b fix(api): correct endpoint name for blend ops 2023-01-15 09:35:25 -06:00
Sean Sube a8f0a7a7eb feat(api): add endpoints for blend and noise settings 2023-01-15 09:32:30 -06:00
Sean Sube e403980a44 feat(api): add parameters for noise source and blend op to inpaint 2023-01-15 09:21:09 -06:00
Sean Sube 77470a610a feat(api): add original source and gaussian blur noise sources 2023-01-15 00:14:05 -06:00
Sean Sube 9376de880e feat(api): add normal and uniform noise sources 2023-01-14 21:46:14 -06:00
Sean Sube 09c9b2c028 fix(api): restore inpainting without outpainting 2023-01-14 18:30:04 -06:00
Sean Sube 34fa3f6341 fix(api): limit outpainting using image size params 2023-01-14 17:09:41 -06:00
Sean Sube c29c92ed90 fix(api): resize images after getting request params 2023-01-14 17:06:29 -06:00
Sean Sube d9bbb9bb5a feat(api): take outpainting dimensions from query params 2023-01-14 16:59:38 -06:00
Sean Sube 899d41a38d run outpainting at original size 2023-01-14 16:31:47 -06:00
Sean Sube 23ddbc6b10 build full-size latents for outpainting 2023-01-14 16:28:43 -06:00
Sean Sube de13d2d6ef remove debug print 2023-01-14 16:27:34 -06:00
Sean Sube 1d4560cf36 fix blend, use mask as mult 2023-01-14 16:24:46 -06:00
Sean Sube 70047737e6 syntax fix 2023-01-14 16:19:54 -06:00
Sean Sube 6772cb4297 per-channel blend 2023-01-14 16:18:37 -06:00
Sean Sube 6fded9de2d build noise by pixels 2023-01-14 16:14:37 -06:00
Sean Sube 4233c34813 syntax fix, save noise image 2023-01-14 16:03:56 -06:00
Sean Sube b8b7dcf757 reshape using image stride 2023-01-14 16:00:12 -06:00
Sean Sube d4d1687851 reshape data before imaging 2023-01-14 15:49:46 -06:00
Sean Sube e05ed35f6b flatten noise data 2023-01-14 15:44:19 -06:00
Sean Sube 1220155d39 use sum of histogram, remove unused imports 2023-01-14 15:36:19 -06:00
Sean Sube e9620fd62e generate noise channel-wise 2023-01-14 15:34:57 -06:00
Sean Sube 0b46ce4090 calculate single stats instance 2023-01-14 15:27:30 -06:00
Sean Sube 9b56c75b92 adjust weights by channel sum 2023-01-14 15:24:04 -06:00
Sean Sube b24b1eb961 feat(gui): produce noise based on source image histogram 2023-01-14 15:19:41 -06:00
Sean Sube c8c548183d multiply source by inverted mask 2023-01-14 14:34:35 -06:00
Sean Sube 5c3d42876f convert pixel back to int 2023-01-14 14:24:36 -06:00
Sean Sube 47df0ecaff convert noise to multiplier 2023-01-14 14:23:31 -06:00
Sean Sube 6216486325 isolate noise blend, print errors 2023-01-14 14:20:42 -06:00
Sean Sube 1e24018b57 feat(api): blend outpainting border with noise 2023-01-14 14:01:15 -06:00
Sean Sube 9ed3a3fd3e move test to inpaint endpoint 2023-01-14 13:24:25 -06:00
Sean Sube a3029c30a6 fix(api): use correct param name for platform 2023-01-14 13:19:04 -06:00
Sean Sube 66dc5322d0 feat(api): add helper to expand images for outpainting 2023-01-14 13:17:16 -06:00
Sean Sube e8b580a5de fix(api): get default params from file, enforce minimum params 2023-01-14 12:45:18 -06:00
Sean Sube fa82ac18ab fix(api): send CORS more consistently 2023-01-14 10:18:53 -06:00
Sean Sube f46647cf6d fix(build): correct path to pip requirements 2023-01-13 23:15:57 -06:00