(self, bucket_id)
| 128 | return hw_id, t_id, ar_id |
| 129 | |
| 130 | def get_thw(self, bucket_id): |
| 131 | assert len(bucket_id) == 3 |
| 132 | T = self.t_criteria[bucket_id[0]][bucket_id[1]] |
| 133 | H, W = self.ar_criteria[bucket_id[0]][bucket_id[1]][bucket_id[2]] |
| 134 | return T, H, W |
| 135 | |
| 136 | def get_prob(self, bucket_id): |
| 137 | return self.bucket_probs[bucket_id[0]][bucket_id[1]] |