AgentDataset返回数据要经过agent.mask文件进行筛选
一下为例
len(zarr_dataset.agents)=1890000
len(AgentDataset)=262304
因为:可以查看mask文件 这里主要对数据进行了晒uxan
import numpy as np
nonzero_count = np.count_nonzero(agents_mask[:, 0])
nonzero_count
Out [9]: 262304
参考文档:https://woven-planet.github.io/l5kit/data_format.html#short-introduction-to-zarr 里面的AgentDataset小节