File size: 6,388 Bytes
7b853a5 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 | # Project Structure
```text
kimodo/
├── kimodo/ # Main Python package
│ ├── model/ # Model architecture and loading
│ │ ├── kimodo_model.py # Kimodo diffusion model wrapper
│ │ ├── twostage_denoiser.py # Two-stage denoising architecture
│ │ ├── backbone.py # Transformer encoder backbone
│ │ ├── diffusion.py # Diffusion process
│ │ ├── cfg.py # Classifier-free guidance
│ │ ├── common.py # Shared model utilities
│ │ ├── load_model.py # Model loading and registry lookup
│ │ ├── loading.py # Checkpoint loading utilities
│ │ ├── registry.py # Model registry (skeleton, checkpoint URLs)
│ │ ├── text_encoder_api.py # Text encoder API client
│ │ ├── tmr.py # TMR compatibility
│ │ └── llm2vec/ # LLM-based text encoder
│ ├── motion_rep/ # Motion representation
│ │ ├── reps/ # Skeleton-specific motion reps
│ │ │ ├── base.py # Base motion rep types
│ │ │ ├── kimodo_motionrep.py
│ │ │ └── tmr_motionrep.py
│ │ ├── conditioning.py # Conditioning (text, constraints)
│ │ ├── feature_utils.py # Feature extraction
│ │ ├── feet.py # Foot contact / smoothing
│ │ ├── smooth_root.py # Smooth root representation
│ │ └── stats.py # Normalization statistics
│ ├── skeleton/ # Skeleton definitions and kinematics
│ │ ├── definitions.py # Skeleton topology (joints, chains)
│ │ ├── registry.py # Skeleton registry
│ │ ├── base.py # Base skeleton types
│ │ ├── kinematics.py # Forward kinematics
│ │ ├── transforms.py # Rotation/transform utilities
│ │ └── bvh.py # BVH I/O
│ ├── viz/ # Visualization
│ │ ├── scene.py # 3D scene setup
│ │ ├── playback.py # Timeline / motion playback
│ │ ├── viser_utils.py # Viser 3D helpers
│ │ ├── gui.py # Demo GUI components
│ │ ├── constraint_ui.py # Constraint editing UI
│ │ ├── coords.py # Coordinate frames
│ │ ├── soma_skin.py # SOMA character skinning
│ │ ├── soma_layer_skin.py # SOMA layer-based skinning
│ │ ├── smplx_skin.py # SMPL-X skinning
│ │ └── g1_rig.py # G1 robot rig
│ ├── demo/ # Interactive web demo
│ │ ├── app.py # Demo entry (Gradio / Viser)
│ │ ├── config.py # Demo configuration
│ │ ├── state.py # Application state
│ │ ├── ui.py # UI layout and callbacks
│ │ ├── generation.py # Generation pipeline for demo
│ │ ├── embedding_cache.py # Cached text embeddings
│ │ ├── queue_manager.py # Request queue for demo
│ │ └── __main__.py # Demo run as module
│ ├── exports/ # Motion export formats
│ │ ├── bvh.py # BVH export
│ │ ├── mujoco.py # MuJoCo export
│ │ └── smplx.py # SMPL-X export
│ ├── metrics/ # Evaluation metrics
│ │ ├── base.py # Metric base classes
│ │ ├── foot_skate.py # Foot skate metric
│ │ ├── constraints.py # Constraint satisfaction
│ │ └── tmr.py # TMR-based metrics
│ ├── scripts/ # CLI and helper scripts
│ │ ├── generate.py # CLI for motion synthesis (kimodo_gen)
│ │ ├── run_text_encoder_server.py # Text encoder server (kimodo_textencoder)
│ │ ├── gradio_theme.py # Gradio theme for demo
│ │ ├── docker-entrypoint.sh # Docker entrypoint for demo
│ │ ├── lock_requirements.py # Dependency locking
│ │ ├── mujoco_load.py # MuJoCo scene loading
│ │ └── ... # Other helpers
│ ├── assets/ # Package data (shipped with package)
│ │ ├── demo/ # Demo examples and config
│ │ └── skeletons/ # Skeleton assets
│ ├── constraints.py # Constraint definitions and handling
│ ├── geometry.py # Geometric utilities
│ ├── postprocess.py # Post-processing (e.g. MotionCorrection)
│ ├── meta.py # Motion metadata
│ ├── sanitize.py # Input sanitization
│ ├── assets.py # Asset path resolution
│ └── tools.py # General utilities
├── MotionCorrection/ # Optional C++/Python post-processing
│ ├── python/motion_correction/ # Python bindings
│ └── src/cpp/ # C++ implementation
├── docs/ # Documentation (Sphinx)
│ └── source/ # RST/MD sources
├── assets/ # Repo-level assets (banner, screenshots)
├── pyproject.toml # Package config and entry points
├── setup.py # Setuptools entry (if needed)
├── Dockerfile # Container image for demo
├── docker-compose.yaml # Docker Compose for demo + text encoder
└── README.md
```
Entry points (from `pyproject.toml`):
- **`kimodo_gen`** — command-line motion synthesis (`kimodo.scripts.generate:main`)
- **`kimodo_demo`** — interactive web demo (`kimodo.demo:main`)
- **`kimodo_textencoder`** — text encoder server (`kimodo.scripts.run_text_encoder_server:main`)
|