(self, idx)
| 90 | return len(self.frame_list) |
| 91 | |
| 92 | def __getitem__(self, idx): |
| 93 | fpath, start_idx = self.frame_list[idx] |
| 94 | demo_traj = np.array(np.load(fpath)) |
| 95 | demo_traj = demo_traj[start_idx:start_idx+self.seq_len] |
| 96 | mask = np.ones(self.seq_len) |
| 97 | demo_size = demo_traj.shape[0] |
| 98 | padding_size = self.seq_len - demo_size |
| 99 | # padded_demo_traj = demo_traj |
| 100 | # # padding |
| 101 | # while padded_demo_traj.shape[0] < self.seq_len: |
| 102 | # mask[padded_demo_traj.shape[0]] *= 0. |
| 103 | # padded_demo_traj = np.concatenate([padded_demo_traj, demo_traj])[:self.seq_len] |
| 104 | # return padded_demo_traj, mask |
| 105 | if padding_size > 0: |
| 106 | demo_traj = np.concatenate([demo_traj, np.zeros((padding_size,)+demo_traj.shape[1:])]) |
| 107 | mask[demo_size-1:] *= 0. |
| 108 | return demo_traj, mask |
| 109 | |
| 110 | |
| 111 |
nothing calls this directly
no outgoing calls
no test coverage detected