add
Browse files- models/backbones/__pycache__/__init__.cpython-310.pyc +0 -0
- models/backbones/__pycache__/vit.cpython-310.pyc +0 -0
- models/modules/ehm/__pycache__/EHM.cpython-310.pyc +0 -0
- models/modules/ehm/__pycache__/EHM_v2.cpython-310.pyc +0 -0
- models/modules/ehm/__pycache__/__init__.cpython-310.pyc +0 -0
- models/modules/flame/__pycache__/FLAME.cpython-310.pyc +0 -0
- models/modules/flame/__pycache__/__init__.cpython-310.pyc +0 -0
- models/modules/flame/__pycache__/lbs.cpython-310.pyc +0 -0
- models/modules/mano/__pycache__/MANO.cpython-310.pyc +0 -0
- models/modules/mano/__pycache__/__init__.cpython-310.pyc +0 -0
- models/modules/renderer/__pycache__/base_renderer.cpython-310.pyc +0 -0
- models/modules/renderer/__pycache__/body_renderer.cpython-310.pyc +0 -0
- models/modules/renderer/__pycache__/util.cpython-310.pyc +0 -0
- models/modules/smplx/__pycache__/SMPLX.cpython-310.pyc +0 -0
- models/modules/smplx/__pycache__/__init__.cpython-310.pyc +0 -0
- models/modules/smplx/__pycache__/lbs.cpython-310.pyc +0 -0
- models/pipeline/__pycache__/ehm_pipeline.cpython-310.pyc +0 -0
- models/pipeline/ehm_pipeline.py +5 -1
- models/smplx/__pycache__/SMPLXV2.cpython-310.pyc +0 -0
- models/smplx/__pycache__/lbs.cpython-310.pyc +0 -0
- models/smplx/__pycache__/pose_transformer.cpython-310.pyc +0 -0
- models/smplx/__pycache__/smplx_head.cpython-310.pyc +0 -0
- models/smplx/__pycache__/smplx_layer.cpython-310.pyc +0 -0
- models/smplx/__pycache__/t_cond_mlp.cpython-310.pyc +0 -0
- utils/__pycache__/__init__.cpython-310.pyc +0 -0
- utils/__pycache__/general_utils.cpython-310.pyc +0 -0
- utils/__pycache__/graphics.cpython-310.pyc +0 -0
- utils/__pycache__/graphics_utils.cpython-310.pyc +0 -0
- utils/__pycache__/helper.cpython-310.pyc +0 -0
- utils/__pycache__/pipeline_utils.cpython-310.pyc +0 -0
- utils/__pycache__/rotation_converter.cpython-310.pyc +0 -0
models/backbones/__pycache__/__init__.cpython-310.pyc
CHANGED
|
Binary files a/models/backbones/__pycache__/__init__.cpython-310.pyc and b/models/backbones/__pycache__/__init__.cpython-310.pyc differ
|
|
|
models/backbones/__pycache__/vit.cpython-310.pyc
CHANGED
|
Binary files a/models/backbones/__pycache__/vit.cpython-310.pyc and b/models/backbones/__pycache__/vit.cpython-310.pyc differ
|
|
|
models/modules/ehm/__pycache__/EHM.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/ehm/__pycache__/EHM.cpython-310.pyc and b/models/modules/ehm/__pycache__/EHM.cpython-310.pyc differ
|
|
|
models/modules/ehm/__pycache__/EHM_v2.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/ehm/__pycache__/EHM_v2.cpython-310.pyc and b/models/modules/ehm/__pycache__/EHM_v2.cpython-310.pyc differ
|
|
|
models/modules/ehm/__pycache__/__init__.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/ehm/__pycache__/__init__.cpython-310.pyc and b/models/modules/ehm/__pycache__/__init__.cpython-310.pyc differ
|
|
|
models/modules/flame/__pycache__/FLAME.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/flame/__pycache__/FLAME.cpython-310.pyc and b/models/modules/flame/__pycache__/FLAME.cpython-310.pyc differ
|
|
|
models/modules/flame/__pycache__/__init__.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/flame/__pycache__/__init__.cpython-310.pyc and b/models/modules/flame/__pycache__/__init__.cpython-310.pyc differ
|
|
|
models/modules/flame/__pycache__/lbs.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/flame/__pycache__/lbs.cpython-310.pyc and b/models/modules/flame/__pycache__/lbs.cpython-310.pyc differ
|
|
|
models/modules/mano/__pycache__/MANO.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/mano/__pycache__/MANO.cpython-310.pyc and b/models/modules/mano/__pycache__/MANO.cpython-310.pyc differ
|
|
|
models/modules/mano/__pycache__/__init__.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/mano/__pycache__/__init__.cpython-310.pyc and b/models/modules/mano/__pycache__/__init__.cpython-310.pyc differ
|
|
|
models/modules/renderer/__pycache__/base_renderer.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/renderer/__pycache__/base_renderer.cpython-310.pyc and b/models/modules/renderer/__pycache__/base_renderer.cpython-310.pyc differ
|
|
|
models/modules/renderer/__pycache__/body_renderer.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/renderer/__pycache__/body_renderer.cpython-310.pyc and b/models/modules/renderer/__pycache__/body_renderer.cpython-310.pyc differ
|
|
|
models/modules/renderer/__pycache__/util.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/renderer/__pycache__/util.cpython-310.pyc and b/models/modules/renderer/__pycache__/util.cpython-310.pyc differ
|
|
|
models/modules/smplx/__pycache__/SMPLX.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/smplx/__pycache__/SMPLX.cpython-310.pyc and b/models/modules/smplx/__pycache__/SMPLX.cpython-310.pyc differ
|
|
|
models/modules/smplx/__pycache__/__init__.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/smplx/__pycache__/__init__.cpython-310.pyc and b/models/modules/smplx/__pycache__/__init__.cpython-310.pyc differ
|
|
|
models/modules/smplx/__pycache__/lbs.cpython-310.pyc
CHANGED
|
Binary files a/models/modules/smplx/__pycache__/lbs.cpython-310.pyc and b/models/modules/smplx/__pycache__/lbs.cpython-310.pyc differ
|
|
|
models/pipeline/__pycache__/ehm_pipeline.cpython-310.pyc
CHANGED
|
Binary files a/models/pipeline/__pycache__/ehm_pipeline.cpython-310.pyc and b/models/pipeline/__pycache__/ehm_pipeline.cpython-310.pyc differ
|
|
|
models/pipeline/ehm_pipeline.py
CHANGED
|
@@ -45,7 +45,11 @@ class Ehm_Pipeline(L.LightningModule):
|
|
| 45 |
# save_image(x[:, :, :, 32:-32],"input.jpg")
|
| 46 |
x = self.normalize(x)
|
| 47 |
B = len(x)
|
| 48 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
| 49 |
# 1. Extract features from image.
|
| 50 |
# The input size is 256*256, but ViT needs 256*192. TODO: make this more elegant.
|
| 51 |
feats = self.backbone(x[:, :, :, 32:-32])
|
|
|
|
| 45 |
# save_image(x[:, :, :, 32:-32],"input.jpg")
|
| 46 |
x = self.normalize(x)
|
| 47 |
B = len(x)
|
| 48 |
+
assert x.numel() > 0
|
| 49 |
+
assert x.shape[0] > 0
|
| 50 |
+
W = x.shape[-1]
|
| 51 |
+
if W <= 64:
|
| 52 |
+
raise ValueError(f"Input width too small: {W}")
|
| 53 |
# 1. Extract features from image.
|
| 54 |
# The input size is 256*256, but ViT needs 256*192. TODO: make this more elegant.
|
| 55 |
feats = self.backbone(x[:, :, :, 32:-32])
|
models/smplx/__pycache__/SMPLXV2.cpython-310.pyc
CHANGED
|
Binary files a/models/smplx/__pycache__/SMPLXV2.cpython-310.pyc and b/models/smplx/__pycache__/SMPLXV2.cpython-310.pyc differ
|
|
|
models/smplx/__pycache__/lbs.cpython-310.pyc
CHANGED
|
Binary files a/models/smplx/__pycache__/lbs.cpython-310.pyc and b/models/smplx/__pycache__/lbs.cpython-310.pyc differ
|
|
|
models/smplx/__pycache__/pose_transformer.cpython-310.pyc
CHANGED
|
Binary files a/models/smplx/__pycache__/pose_transformer.cpython-310.pyc and b/models/smplx/__pycache__/pose_transformer.cpython-310.pyc differ
|
|
|
models/smplx/__pycache__/smplx_head.cpython-310.pyc
CHANGED
|
Binary files a/models/smplx/__pycache__/smplx_head.cpython-310.pyc and b/models/smplx/__pycache__/smplx_head.cpython-310.pyc differ
|
|
|
models/smplx/__pycache__/smplx_layer.cpython-310.pyc
CHANGED
|
Binary files a/models/smplx/__pycache__/smplx_layer.cpython-310.pyc and b/models/smplx/__pycache__/smplx_layer.cpython-310.pyc differ
|
|
|
models/smplx/__pycache__/t_cond_mlp.cpython-310.pyc
CHANGED
|
Binary files a/models/smplx/__pycache__/t_cond_mlp.cpython-310.pyc and b/models/smplx/__pycache__/t_cond_mlp.cpython-310.pyc differ
|
|
|
utils/__pycache__/__init__.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/__init__.cpython-310.pyc and b/utils/__pycache__/__init__.cpython-310.pyc differ
|
|
|
utils/__pycache__/general_utils.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/general_utils.cpython-310.pyc and b/utils/__pycache__/general_utils.cpython-310.pyc differ
|
|
|
utils/__pycache__/graphics.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/graphics.cpython-310.pyc and b/utils/__pycache__/graphics.cpython-310.pyc differ
|
|
|
utils/__pycache__/graphics_utils.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/graphics_utils.cpython-310.pyc and b/utils/__pycache__/graphics_utils.cpython-310.pyc differ
|
|
|
utils/__pycache__/helper.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/helper.cpython-310.pyc and b/utils/__pycache__/helper.cpython-310.pyc differ
|
|
|
utils/__pycache__/pipeline_utils.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/pipeline_utils.cpython-310.pyc and b/utils/__pycache__/pipeline_utils.cpython-310.pyc differ
|
|
|
utils/__pycache__/rotation_converter.cpython-310.pyc
CHANGED
|
Binary files a/utils/__pycache__/rotation_converter.cpython-310.pyc and b/utils/__pycache__/rotation_converter.cpython-310.pyc differ
|
|
|