├── .gitignore ├── LICENSE ├── README.md ├── evaluation_metrics.py ├── manip ├── data │ ├── bps.pt │ └── hand_foot_dataset.py ├── lafan1 │ └── utils.py ├── model │ ├── transformer_fullbody_cond_diffusion_model.py │ ├── transformer_hand_foot_manip_cond_diffusion_model.py │ └── transformer_module.py └── vis │ ├── blender_vis_human_utils.py │ ├── blender_vis_mesh_motion.py │ ├── blender_vis_object_utils.py │ ├── blender_vis_utils.py │ └── floor_colorful_mat.blend ├── omomo_teaser.png ├── omomo_text_anno.zip ├── requirements.txt ├── scriprts ├── test_omomo.sh ├── train_stage1.sh └── train_stage2.sh ├── trainer_full_body_manip_diffusion.py └── trainer_hand_foot_manip_diffusion.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/README.md -------------------------------------------------------------------------------- /evaluation_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/evaluation_metrics.py -------------------------------------------------------------------------------- /manip/data/bps.pt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/data/bps.pt -------------------------------------------------------------------------------- /manip/data/hand_foot_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/data/hand_foot_dataset.py -------------------------------------------------------------------------------- /manip/lafan1/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/lafan1/utils.py -------------------------------------------------------------------------------- /manip/model/transformer_fullbody_cond_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/model/transformer_fullbody_cond_diffusion_model.py -------------------------------------------------------------------------------- /manip/model/transformer_hand_foot_manip_cond_diffusion_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/model/transformer_hand_foot_manip_cond_diffusion_model.py -------------------------------------------------------------------------------- /manip/model/transformer_module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/model/transformer_module.py -------------------------------------------------------------------------------- /manip/vis/blender_vis_human_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/vis/blender_vis_human_utils.py -------------------------------------------------------------------------------- /manip/vis/blender_vis_mesh_motion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/vis/blender_vis_mesh_motion.py -------------------------------------------------------------------------------- /manip/vis/blender_vis_object_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/vis/blender_vis_object_utils.py -------------------------------------------------------------------------------- /manip/vis/blender_vis_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/vis/blender_vis_utils.py -------------------------------------------------------------------------------- /manip/vis/floor_colorful_mat.blend: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/manip/vis/floor_colorful_mat.blend -------------------------------------------------------------------------------- /omomo_teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/omomo_teaser.png -------------------------------------------------------------------------------- /omomo_text_anno.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/omomo_text_anno.zip -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/requirements.txt -------------------------------------------------------------------------------- /scriprts/test_omomo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/scriprts/test_omomo.sh -------------------------------------------------------------------------------- /scriprts/train_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/scriprts/train_stage1.sh -------------------------------------------------------------------------------- /scriprts/train_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/scriprts/train_stage2.sh -------------------------------------------------------------------------------- /trainer_full_body_manip_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/trainer_full_body_manip_diffusion.py -------------------------------------------------------------------------------- /trainer_hand_foot_manip_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/lijiaman/omomo_release/HEAD/trainer_hand_foot_manip_diffusion.py --------------------------------------------------------------------------------