├── .gitignore ├── LICENSE ├── README.md ├── configs ├── __init__.py ├── default.yaml └── yacs │ ├── LICENSE │ ├── NOTICE │ ├── __init__.py │ └── yacs.py ├── data ├── mdm_poses │ └── sample.npy └── snapshot │ └── poses │ ├── female-3-casual │ └── poses │ │ ├── anim_nerf_test.npz │ │ ├── anim_nerf_train.npz │ │ └── anim_nerf_val.npz │ ├── female-4-casual │ └── poses │ │ ├── anim_nerf_test.npz │ │ ├── anim_nerf_train.npz │ │ └── anim_nerf_val.npz │ ├── male-3-casual │ └── poses │ │ ├── anim_nerf_test.npz │ │ ├── anim_nerf_train.npz │ │ └── anim_nerf_val.npz │ └── male-4-casual │ └── poses │ ├── anim_nerf_test.npz │ ├── anim_nerf_train.npz │ └── anim_nerf_val.npz ├── dataset ├── freeview.py ├── newpose.py ├── test.py └── train.py ├── eval.py ├── exps ├── snapshot_f3c.yaml ├── snapshot_f4c.yaml ├── snapshot_m3c.yaml ├── snapshot_m4c.yaml ├── zju-mocap_377.yaml ├── zju-mocap_386.yaml ├── zju-mocap_387.yaml ├── zju-mocap_392.yaml ├── zju-mocap_393.yaml └── zju-mocap_394.yaml ├── models ├── model.py └── modules │ ├── appearance_module.py │ ├── non_rigid_module.py │ ├── pose_refinement_module.py │ ├── renderer │ ├── __init__.py │ ├── gaussian.py │ └── mesh.py │ └── shadow_module.py ├── requirements.txt ├── scripts ├── prepare_snapshot │ ├── female-3-casual.yaml │ ├── female-3-casual_test.yaml │ ├── female-4-casual.yaml │ ├── female-4-casual_test.yaml │ ├── male-3-casual.yaml │ ├── male-3-casual_test.yaml │ ├── male-4-casual.yaml │ ├── male-4-casual_test.yaml │ └── prepare_dataset.py └── prepare_zju-mocap │ ├── 377.yaml │ ├── 386.yaml │ ├── 387.yaml │ ├── 392.yaml │ ├── 393.yaml │ ├── 394.yaml │ └── prepare_dataset.py ├── train.py ├── train_pose.py └── utils ├── __init__.py ├── body_util.py ├── camera_util.py ├── file_util.py ├── image_util.py ├── lpips ├── LICENSE ├── __init__.py ├── lpips.py ├── pretrained_networks.py └── weights │ └── v0.1 │ └── vgg.pth ├── network_util.py ├── pc_util.py ├── smpl └── smpl_numpy.py ├── tb_util.py └── train_util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/README.md -------------------------------------------------------------------------------- /configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/configs/__init__.py -------------------------------------------------------------------------------- /configs/default.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/configs/default.yaml -------------------------------------------------------------------------------- /configs/yacs/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/configs/yacs/LICENSE -------------------------------------------------------------------------------- /configs/yacs/NOTICE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/configs/yacs/NOTICE -------------------------------------------------------------------------------- /configs/yacs/__init__.py: -------------------------------------------------------------------------------- 1 | from .yacs import CfgNode -------------------------------------------------------------------------------- /configs/yacs/yacs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/configs/yacs/yacs.py -------------------------------------------------------------------------------- /data/mdm_poses/sample.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/mdm_poses/sample.npy -------------------------------------------------------------------------------- /data/snapshot/poses/female-3-casual/poses/anim_nerf_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/female-3-casual/poses/anim_nerf_test.npz -------------------------------------------------------------------------------- /data/snapshot/poses/female-3-casual/poses/anim_nerf_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/female-3-casual/poses/anim_nerf_train.npz -------------------------------------------------------------------------------- /data/snapshot/poses/female-3-casual/poses/anim_nerf_val.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/female-3-casual/poses/anim_nerf_val.npz -------------------------------------------------------------------------------- /data/snapshot/poses/female-4-casual/poses/anim_nerf_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/female-4-casual/poses/anim_nerf_test.npz -------------------------------------------------------------------------------- /data/snapshot/poses/female-4-casual/poses/anim_nerf_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/female-4-casual/poses/anim_nerf_train.npz -------------------------------------------------------------------------------- /data/snapshot/poses/female-4-casual/poses/anim_nerf_val.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/female-4-casual/poses/anim_nerf_val.npz -------------------------------------------------------------------------------- /data/snapshot/poses/male-3-casual/poses/anim_nerf_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/male-3-casual/poses/anim_nerf_test.npz -------------------------------------------------------------------------------- /data/snapshot/poses/male-3-casual/poses/anim_nerf_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/male-3-casual/poses/anim_nerf_train.npz -------------------------------------------------------------------------------- /data/snapshot/poses/male-3-casual/poses/anim_nerf_val.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/male-3-casual/poses/anim_nerf_val.npz -------------------------------------------------------------------------------- /data/snapshot/poses/male-4-casual/poses/anim_nerf_test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/male-4-casual/poses/anim_nerf_test.npz -------------------------------------------------------------------------------- /data/snapshot/poses/male-4-casual/poses/anim_nerf_train.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/male-4-casual/poses/anim_nerf_train.npz -------------------------------------------------------------------------------- /data/snapshot/poses/male-4-casual/poses/anim_nerf_val.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/data/snapshot/poses/male-4-casual/poses/anim_nerf_val.npz -------------------------------------------------------------------------------- /dataset/freeview.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/dataset/freeview.py -------------------------------------------------------------------------------- /dataset/newpose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/dataset/newpose.py -------------------------------------------------------------------------------- /dataset/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/dataset/test.py -------------------------------------------------------------------------------- /dataset/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/dataset/train.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/eval.py -------------------------------------------------------------------------------- /exps/snapshot_f3c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/snapshot_f3c.yaml -------------------------------------------------------------------------------- /exps/snapshot_f4c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/snapshot_f4c.yaml -------------------------------------------------------------------------------- /exps/snapshot_m3c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/snapshot_m3c.yaml -------------------------------------------------------------------------------- /exps/snapshot_m4c.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/snapshot_m4c.yaml -------------------------------------------------------------------------------- /exps/zju-mocap_377.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/zju-mocap_377.yaml -------------------------------------------------------------------------------- /exps/zju-mocap_386.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/zju-mocap_386.yaml -------------------------------------------------------------------------------- /exps/zju-mocap_387.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/zju-mocap_387.yaml -------------------------------------------------------------------------------- /exps/zju-mocap_392.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/zju-mocap_392.yaml -------------------------------------------------------------------------------- /exps/zju-mocap_393.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/zju-mocap_393.yaml -------------------------------------------------------------------------------- /exps/zju-mocap_394.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/exps/zju-mocap_394.yaml -------------------------------------------------------------------------------- /models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/model.py -------------------------------------------------------------------------------- /models/modules/appearance_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/appearance_module.py -------------------------------------------------------------------------------- /models/modules/non_rigid_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/non_rigid_module.py -------------------------------------------------------------------------------- /models/modules/pose_refinement_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/pose_refinement_module.py -------------------------------------------------------------------------------- /models/modules/renderer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/renderer/__init__.py -------------------------------------------------------------------------------- /models/modules/renderer/gaussian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/renderer/gaussian.py -------------------------------------------------------------------------------- /models/modules/renderer/mesh.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/renderer/mesh.py -------------------------------------------------------------------------------- /models/modules/shadow_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/models/modules/shadow_module.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/prepare_snapshot/female-3-casual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/female-3-casual.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/female-3-casual_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/female-3-casual_test.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/female-4-casual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/female-4-casual.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/female-4-casual_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/female-4-casual_test.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/male-3-casual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/male-3-casual.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/male-3-casual_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/male-3-casual_test.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/male-4-casual.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/male-4-casual.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/male-4-casual_test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/male-4-casual_test.yaml -------------------------------------------------------------------------------- /scripts/prepare_snapshot/prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_snapshot/prepare_dataset.py -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/377.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/377.yaml -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/386.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/386.yaml -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/387.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/387.yaml -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/392.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/392.yaml -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/393.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/393.yaml -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/394.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/394.yaml -------------------------------------------------------------------------------- /scripts/prepare_zju-mocap/prepare_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/scripts/prepare_zju-mocap/prepare_dataset.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/train.py -------------------------------------------------------------------------------- /train_pose.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/train_pose.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/body_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/body_util.py -------------------------------------------------------------------------------- /utils/camera_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/camera_util.py -------------------------------------------------------------------------------- /utils/file_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/file_util.py -------------------------------------------------------------------------------- /utils/image_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/image_util.py -------------------------------------------------------------------------------- /utils/lpips/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/lpips/LICENSE -------------------------------------------------------------------------------- /utils/lpips/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/lpips/__init__.py -------------------------------------------------------------------------------- /utils/lpips/lpips.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/lpips/lpips.py -------------------------------------------------------------------------------- /utils/lpips/pretrained_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/lpips/pretrained_networks.py -------------------------------------------------------------------------------- /utils/lpips/weights/v0.1/vgg.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/lpips/weights/v0.1/vgg.pth -------------------------------------------------------------------------------- /utils/network_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/network_util.py -------------------------------------------------------------------------------- /utils/pc_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/pc_util.py -------------------------------------------------------------------------------- /utils/smpl/smpl_numpy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/smpl/smpl_numpy.py -------------------------------------------------------------------------------- /utils/tb_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/tb_util.py -------------------------------------------------------------------------------- /utils/train_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wenj/GoMAvatar/HEAD/utils/train_util.py --------------------------------------------------------------------------------