411 Commits

Author	SHA1	Message	Date
Sean Sube	3e5a95548b	fix array sizing, update release tests	2024-01-06 02:33:01 -06:00
Sean Sube	1a8d538bfe	handle empty results	2024-01-05 21:23:05 -06:00
Sean Sube	058a0f1c0a	use max size of all images in stack	2024-01-05 21:19:48 -06:00
Sean Sube	b5ed1101b4	use result size helper	2024-01-05 21:16:10 -06:00
Sean Sube	e958eff3a8	misc testing fixes	2024-01-05 21:15:33 -06:00
Sean Sube	7d782842ed	fix(api): only hash networks once per worker lifetime	2024-01-05 21:11:44 -06:00
Sean Sube	88f99ef6c2	remove unused format field from upscale params	2024-01-05 20:13:16 -06:00
Sean Sube	9b5d00a66a	move progress state to worker context	2024-01-05 20:12:41 -06:00
Sean Sube	4f230f4111	clean up result and metadata handling	2024-01-05 20:11:58 -06:00
Sean Sube	10acad232c	estimate steps better, pass progress onto reply	2024-01-04 19:48:43 -06:00
Sean Sube	4dc251cf4a	fix callback access	2024-01-04 19:39:44 -06:00
Sean Sube	9a1389d03b	fix imports	2024-01-04 19:16:40 -06:00
Sean Sube	b6da935be6	use progress type in command	2024-01-04 19:09:52 -06:00
Sean Sube	ce84dfa115	pass metadata through tiles	2024-01-04 09:12:32 -06:00
Sean Sube	112d5a8876	start passing metadata through stages	2024-01-04 09:01:19 -06:00
Sean Sube	48671e27b0	fix hash field	2024-01-03 23:46:50 -06:00
Sean Sube	fe6b267483	prep models field	2024-01-03 23:38:44 -06:00
Sean Sube	bdefaec0d2	serialize metadata	2024-01-03 23:27:40 -06:00
Sean Sube	fac98ab239	keep result of each stage with metadata	2024-01-03 23:13:21 -06:00
Sean Sube	e0d0933092	fix missing param	2024-01-03 22:20:27 -06:00
Sean Sube	46bcd5af86	make response types match	2024-01-03 22:15:50 -06:00
Sean Sube	28fc2082c7	track results after each stage	2024-01-03 21:31:41 -06:00
Sean Sube	4f1bc84fd9	initialize fields	2024-01-03 21:20:46 -06:00
Sean Sube	0c504e3f69	report stage and tile count in progress	2024-01-03 21:16:44 -06:00
Sean Sube	44a8d61082	feat: add batch endpoints for cancel and status, update responses	2024-01-03 19:09:18 -06:00
Sean Sube	19c91f70f5	add networks to metadata	2024-01-02 22:14:21 -06:00
Sean Sube	46098960d8	attempt to save grid mode metadata	2024-01-02 21:49:22 -06:00
Sean Sube	297285fb6f	fix(api): upscale during highres regardless of upscale checkbox	2024-01-02 21:34:11 -06:00
Sean Sube	4edd39740b	add image metadata to stage results	2024-01-02 21:24:27 -06:00
Sean Sube	de8ab6f1c9	fix(api): use correct part of the gfpgan output	2023-12-31 07:04:43 -06:00
Sean Sube	c6718a3e3b	fix(api): make sure upscaling only runs once regardless of correction options	2023-12-31 06:39:00 -06:00
Sean Sube	71fbc871a6	feat: add API parameter for upscale checkbox	2023-12-30 19:34:40 -06:00
Sean Sube	7abe6dc6a9	feat(api): enable 1x upscaling models	2023-12-30 11:59:52 -06:00
Sean Sube	f6134dafc6	fix errors in local std denoising	2023-12-29 23:19:35 -06:00
Sean Sube	1035915d36	apply lint	2023-12-29 08:19:58 -06:00
Sean Sube	ce90ffb0ee	use threshold range for local denoising	2023-12-29 08:18:37 -06:00
Sean Sube	f28fdda47a	feat(api): add stage for local standard deviation denoising for XL	2023-12-27 20:17:35 -06:00
Sean Sube	7d56689527	pass device to wrapper	2023-12-27 09:01:38 -06:00
Sean Sube	d17b946091	wrap ESRGAN helper	2023-12-27 08:47:06 -06:00
Sean Sube	cfe7a55935	type and test fixes	2023-12-26 20:21:34 -06:00
Sean Sube	2b65077d82	type lint	2023-12-26 08:06:16 -06:00
Sean Sube	3d245a663d	fix(api): correct origin point when blending images (#445)	2023-12-22 11:31:31 -06:00
Sean Sube	1fbee0ae52	fix(api): tile stage masks	2023-12-20 23:33:13 -06:00
Sean Sube	f6ff53d0d1	pass correct crop	2023-12-20 23:02:58 -06:00
Sean Sube	de28ecdfad	fix(api): handle blending images of different sizes better	2023-12-20 22:58:57 -06:00
Sean Sube	6e5d10f521	apply lint	2023-12-18 22:04:24 -06:00
Sean Sube	2263093c4d	convert images to cv2	2023-12-18 21:53:20 -06:00
Sean Sube	3e1db707ac	fix(api): update codeformer patches for new lib	2023-12-18 21:12:08 -06:00
Sean Sube	7ed30ee470	feat(api): switch to codeformer lib that works with torch 2.x	2023-12-18 21:06:39 -06:00
Sean Sube	6c4f4f334f	always use alpha in blend stage	2023-12-14 20:12:39 -06:00