BestWJH commited on
Commit
d5f659e
·
1 Parent(s): 84e4a66
Files changed (31) hide show
  1. models/backbones/__pycache__/__init__.cpython-310.pyc +0 -0
  2. models/backbones/__pycache__/vit.cpython-310.pyc +0 -0
  3. models/modules/ehm/__pycache__/EHM.cpython-310.pyc +0 -0
  4. models/modules/ehm/__pycache__/EHM_v2.cpython-310.pyc +0 -0
  5. models/modules/ehm/__pycache__/__init__.cpython-310.pyc +0 -0
  6. models/modules/flame/__pycache__/FLAME.cpython-310.pyc +0 -0
  7. models/modules/flame/__pycache__/__init__.cpython-310.pyc +0 -0
  8. models/modules/flame/__pycache__/lbs.cpython-310.pyc +0 -0
  9. models/modules/mano/__pycache__/MANO.cpython-310.pyc +0 -0
  10. models/modules/mano/__pycache__/__init__.cpython-310.pyc +0 -0
  11. models/modules/renderer/__pycache__/base_renderer.cpython-310.pyc +0 -0
  12. models/modules/renderer/__pycache__/body_renderer.cpython-310.pyc +0 -0
  13. models/modules/renderer/__pycache__/util.cpython-310.pyc +0 -0
  14. models/modules/smplx/__pycache__/SMPLX.cpython-310.pyc +0 -0
  15. models/modules/smplx/__pycache__/__init__.cpython-310.pyc +0 -0
  16. models/modules/smplx/__pycache__/lbs.cpython-310.pyc +0 -0
  17. models/pipeline/__pycache__/ehm_pipeline.cpython-310.pyc +0 -0
  18. models/pipeline/ehm_pipeline.py +5 -1
  19. models/smplx/__pycache__/SMPLXV2.cpython-310.pyc +0 -0
  20. models/smplx/__pycache__/lbs.cpython-310.pyc +0 -0
  21. models/smplx/__pycache__/pose_transformer.cpython-310.pyc +0 -0
  22. models/smplx/__pycache__/smplx_head.cpython-310.pyc +0 -0
  23. models/smplx/__pycache__/smplx_layer.cpython-310.pyc +0 -0
  24. models/smplx/__pycache__/t_cond_mlp.cpython-310.pyc +0 -0
  25. utils/__pycache__/__init__.cpython-310.pyc +0 -0
  26. utils/__pycache__/general_utils.cpython-310.pyc +0 -0
  27. utils/__pycache__/graphics.cpython-310.pyc +0 -0
  28. utils/__pycache__/graphics_utils.cpython-310.pyc +0 -0
  29. utils/__pycache__/helper.cpython-310.pyc +0 -0
  30. utils/__pycache__/pipeline_utils.cpython-310.pyc +0 -0
  31. 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