Repository navigation
Releases: ToberJ/reflex
Release list
Reflex v1.0.0 — code and checkpoints
Released checkpoints of Reflex (the ones evaluated in the paper). scripts/download_checkpoints.sh fetches and verifies them against assets/checkpoints.sha256.
| File | Stage |
|---|---|
teacher_stage1.pt |
1 — state-based teacher (PPO + AMP) |
bc_init_stage2.pt |
2 — behaviour-cloning initialisation of encoder + decoder |
latent_stage2.pt |
2 — latent controller after joint PPO |
stage3_heatmap_cnn.pt, stage3_feature_net.pt, stage3_latent_weights.npz, visual_encoder_stage3.pt |
3 — egocentric RGB-D visual encoder and its intermediate steps |
The Stage 3 rendered frame set (2.1 GB) is in the separate release stage3-frames-v1 (scripts/download_stage3_frames.sh). See docs/REPRODUCE.md for what each file reproduces.
Stage 3 frame set (96k RGB-D frames, 2.1 GB)
Stage 3 rendered frame set: 96,000 egocentric RGB-D frames (192 × 144) with box labels, used by step 1 of scripts/train_stage3.sh (heatmap CNN). 13 files, 2.1 GB; scripts/download_stage3_frames.sh fetches them into data/stage3/frames/v10_s7_rgbd/ and verifies assets/stage3_frames.sha256. Format and provenance: docs/STAGE3.md.
Not needed for evaluation, nor for Stage 3 training started from the released CNN (--cnn data/checkpoints/stage3_heatmap_cnn.pt).