verityw commited on
Commit
ac2c562
·
1 Parent(s): 7bff5a4

update readme with filter info

Browse files
Files changed (1) hide show
  1. README.md +9 -0
README.md CHANGED
@@ -74,6 +74,15 @@ pixel_positions = intrinsics_matrix @ robot_gripper_position_cam
74
  pixel_positions = pixel_positions[:2] / pixel_positions[2] # Shape 2 x 1 # Done!
75
  ```
76
 
 
 
 
 
 
 
 
 
 
77
  ## Accessing Annotation Data
78
 
79
  All annotations are stored in `json` files which you can download from this repository.
 
74
  pixel_positions = pixel_positions[:2] / pixel_positions[2] # Shape 2 x 1 # Done!
75
  ```
76
 
77
+ ## Filtering Data
78
+ Many episodes in DROID contain significant pauses. This is an issue when training models, as these pauses typically happen at the start of episodes, causing the policy to likewise output idle actions when in the home position. To remediate this, we recommend filtering the data you train your policy on, removing all frames that map to idle actions.
79
+
80
+ We provide `keep_ranges_1_0_1.json` which maps episode keys to a list of time step ranges that should *not* be filtered out. The episode keys uniquely identify each episode, and are defined as `f"{recording_folderpath}--{file_path}"`. We opt for this unique identifier because both pieces of information are found in the episodes' RLDS metadata, and thus is easy to compute (even with TensorFlow symbolic operations).
81
+
82
+ To use this data, we recommend creating a `tf.lookup.StaticHashTable` identifying all frames that should not be filtered (with all other frames being filtered by default). Frames can be uniquely identified by simply concatenating their episode key with their time step within the episode.
83
+
84
+ This particular filter `json` is meant for `droid/1.0.1`, NOT `droid/1.0.0`. It was computed by finding all continuous sequences in episodes of non-idle actions that are at least of length 16 (1 second of wallclock time) that are not interrupted by 8 or more idle actions.
85
+
86
  ## Accessing Annotation Data
87
 
88
  All annotations are stored in `json` files which you can download from this repository.