Skip to content

Releases: ToberJ/reflex

Reflex v1.0.0 — code and checkpoints

Choose a tag to compare

@ToberJ ToberJ released this 13 Oct 13:15

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)

Choose a tag to compare

@ToberJ ToberJ released this 13 Oct 13:15

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).