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