Function get_patch_embed

dust3r/patch_embed.py:13–16 · view source on GitHub ↗

(patch_embed_cls, img_size, patch_size, enc_embed_dim)

Source from the content-addressed store, hash-verified

11
12
13	def get_patch_embed(patch_embed_cls, img_size, patch_size, enc_embed_dim):
14	assert patch_embed_cls in ['PatchEmbedDust3R', 'ManyAR_PatchEmbed']
15	patch_embed = eval(patch_embed_cls)(img_size, patch_size, 3, enc_embed_dim)
16	return patch_embed
17
18
19	class PatchEmbedDust3R(PatchEmbed):

_set_patch_embedMethod · 0.90

no outgoing calls

no test coverage detected