diffusers
[Pipeline] Marigold depth and normals estimation
#7847
Merged

[Pipeline] Marigold depth and normals estimation #7847

toshas
toshas implement marigold depth and normals pipelines in diffusers core
32d5131f
sayakpaul sayakpaul requested a review from yiyixuxu yiyixuxu 2 years ago
sayakpaul sayakpaul requested a review from sayakpaul sayakpaul 2 years ago
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
sayakpaul
sayakpaul commented on 2024-05-03
yiyixuxu
yiyixuxu commented on 2024-05-03
toshas remove bibtex
23255c2f
toshas Merge branch 'main' into pipeline_marigold_pr
59ec93b1
toshas remove deprecations
4bde96cc
toshas remove save_memory argument
f12d94cb
toshas remove validate_vae
bf877895
toshas remove config output
bce8dd68
toshas remove batch_size autodetection
b2990115
toshas remove presets logic
e9d6bdfa
toshas remove no_grad
9750185c
toshas add fp16 to the example usage
ef881c65
toshas implement is_matplotlib_available
c1c7520f
toshas move colormap, visualize_depth, and visualize_normals into export_uti…
04c807d7
toshas make the denoising loop more lucid
3de290f3
toshas style
a013e2fb
toshas
yiyixuxu
yiyixuxu commented on 2024-05-12
toshas Merge branch 'main' into pipeline_marigold_pr
142df365
toshas rename denoising_steps into num_inference_steps
2e7e747f
toshas rename input_image into image
405feb31
toshas rename input_latent into latents
b72334b5
toshas remove decode_image
0878b3ef
toshas move clean_latent outside of progress_bar
601bfc8b
toshas refactor marigold-reusable image processing bits into MarigoldImagePr…
a21da454
yiyixuxu
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
sayakpaul
sayakpaul commented on 2024-05-14
HuggingFaceDocBuilderDev
yiyixuxu
yiyixuxu commented on 2024-05-14
yiyixuxu
yiyixuxu commented on 2024-05-14
toshas Merge branch 'main' into pipeline_marigold_pr
6b48618f
toshas clean up the usage example docstring
5847bbd0
toshas make ensemble functions members of the pipelines
a1fd5515
toshas add early checks in check_inputs
6f17e2cc
toshas fix vae_scale_factor computation
832cbd85
toshas better compatibility with torch.compile
c4ff406a
toshas move export_depth_to_png to export_utils
f3ae063f
toshas remove encode_prediction
706ff6d1
toshas improve visualize_depth and visualize_normals to accept multi-dimensi…
b699d76a
toshas do not shortcut vae.config variables
c72c5b7f
toshas change all asserts to raise ValueError
0d09a0cc
toshas rename output_prediction_type to output_type
b597ee89
toshas better variable names
91cba308
toshas better variable names
99213c36
toshas pass desc and leave kwargs into the diffusers progress_bar
bca2571b
toshas implement scale_invariant and shift_invariant flags in the ensemble_d…
30a15fa5
toshas fix generator device placement checks
2b276f6c
toshas move encode_empty_text body into the pipeline call
d8a2550a
toshas minor empty text encoding simplifications
24ff8964
toshas adjust pipelines' class docstrings to explain the added construction …
d9f9ca58
toshas improve the scipy failure condition
ee56c355
toshas make input image values range check configurable in the preprocessor
1b2c027c
toshas remove forgotten print
a06bf1f5
toshas add prediction_type model config
2b3261ee
toshas add uncertainty visualization into export utils
0627efc0
toshas change default of output_uncertainty to False
7c16b078
toshas Merge branch 'main' into pipeline_marigold_pr
9767ddd3
yiyixuxu
yiyixuxu commented on 2024-05-20
toshas fix `output_uncertainty=False`
6e6e1326
toshas remove kwargs
3634eb06
toshas rename prepare_latent into prepare_latents as in other pipelines
f4813c70
toshas move nested-capable `progress_bar` method into the pipelines
566e8949
toshas minor message improvement
b1a151ed
toshas fix cpu offloading
03eb96e7
yiyixuxu
yiyixuxu commented on 2024-05-20
yiyixuxu
yiyixuxu commented on 2024-05-20
toshas move colormap, visualize_depth, export_depth_to_16bit_png, visualize_…
f4845bad
toshas fix missing comma
b5052e36
toshas change torch.FloatTensor to torch.Tensor
0f308bbc
toshas fix importing of MarigoldImageProcessor
9db8c177
toshas fix vae offloading
c56d2fa1
yiyixuxu
yiyixuxu commented on 2024-05-21
toshas implement marigold's intial tests
f5cfeafa
toshas fix num_images computation
662e1d5a
toshas remove MarigoldImageProcessor and outputs from import structure
f03e27d1
toshas update docstrings
6507f1b5
yiyixuxu
yiyixuxu commented on 2024-05-21
yiyixuxu
yiyixuxu
yiyixuxu commented on 2024-05-23
yiyixuxu update init
596ada31
yiyixuxu update
1b89bf66
style
9814973d
yiyixuxu fix
f2a74acb
yiyixuxu fix
b35c6ef9
up
412aafc0
yiyixuxu up
64978027
up
a4d321c1
yiyixuxu add simple test
a778f4d8
up
d0da66c6
update expected np input/output to be channel last
107c1da0
toshas move expand_tensor_or_array into the MarigoldImageProcessor
2345f6e5
toshas rewrite tests to follow conventions - hardcoded slices instead of ima…
b3f3c88f
toshas Merge branch 'main' into pipeline_marigold_pr
80dcdf57
sayakpaul add basic docs.
16e07a14
sayakpaul add anton's contribution statement
4aa8ce54
sayakpaul Merge branch 'main' into pipeline_marigold_pr
0b495c1a
sayakpaul remove todos.
a2df838d
sayakpaul fix assertion values for marigold depth slow tests
00082c89
sayakpaul fix assertion values for depth normals.
09276fbf
sayakpaul remove print
c044f2e1
sayakpaul Merge branch 'main' into pipeline_marigold_pr
699976cb
toshas support AutoencoderTiny in the pipelines
b0fc7f36
toshas update documentation page
10fc97dc
toshas fix missing import in docstring
6e053b40
toshas [doc] add marigold to pipelines overview
56f7980d
toshas [doc] add section "usage examples"
2cbe30b5
toshas fix an issue with latents check in the pipelines
35925b96
toshas add "Frame-by-frame Video Processing with Consistency" section
f808e880
toshas grammarly
183327a5
toshas replace tables with images with css-styled images (blindly)
4b286f85
toshas
sayakpaul Merge branch 'main' into pipeline_marigold_pr
59fe25cb
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul style
c587f5d1
sayakpaul print
9dc2f317
sayakpaul fix the assertions.
304ab64f
sayakpaul Merge branch 'main' into pipeline_marigold_pr
ab8d5cef
sayakpaul take from the github runner.
04941312
sayakpaul take the slices from action artifacts
bfaf0023
sayakpaul style.
ad3e4844
sayakpaul update with the slices from the runner.
cab05075
sayakpaul remove unnecessary code blocks.
427637cb
toshas Merge branch 'main' into pipeline_marigold_pr
037fa29c
toshas Revert "[doc] add marigold to pipelines overview"
2da329e7
toshas remove invitation for new modalities
bdd21be2
toshas split out marigold usage examples
38157781
toshas
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul commented on 2024-05-27
sayakpaul
sayakpaul approved these changes on 2024-05-27
sayakpaul doc cleanup
20b08ba6
sayakpaul Merge branch 'main' into pipeline_marigold_pr
8aae163e
sayakpaul sayakpaul merged b3d10d6d into main 2 years ago
sayakpaul
toshas

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
Milestone