Inherited from the V2XSimSeg dataset, now support multi timestamp loading output padded_voxel_points_next are for other timestamp it is an empty list if timestamp = 1
(
self,
dataset_roots=None,
config=None,
split=None,
cache_size=1000,
val=False,
com=False,
bound=None,
kd_flag=False,
no_cross_road=False,
time_stamp = 2,
)
| 46 | |
| 47 | class MultiTempV2XSimSeg(V2XSimSeg): |
| 48 | def __init__( |
| 49 | self, |
| 50 | dataset_roots=None, |
| 51 | config=None, |
| 52 | split=None, |
| 53 | cache_size=1000, |
| 54 | val=False, |
| 55 | com=False, |
| 56 | bound=None, |
| 57 | kd_flag=False, |
| 58 | no_cross_road=False, |
| 59 | time_stamp = 2, |
| 60 | ): |
| 61 | """ |
| 62 | Inherited from the V2XSimSeg dataset, now support multi timestamp loading |
| 63 | output padded_voxel_points_next are for other timestamp |
| 64 | it is an empty list if timestamp = 1 |
| 65 | """ |
| 66 | super().__init__(dataset_roots, config, split, cache_size, val, com, bound, kd_flag, no_cross_road) |
| 67 | self.time_stamp = config.time_stamp if hasattr(config, 'time_stamp') else time_stamp |
| 68 | print("use time stamp", self.time_stamp) |
| 69 | |
| 70 | def get_seginfo_from_single_agent(self, agent_id, idx): |
| 71 | empty_flag = False |