数据加载与处理 (Dataset)

Submodules

• src.dataset.base_dataset module
  • RasterizedMap
    • RasterizedMap.map
    • RasterizedMap.xmin
    • RasterizedMap.ymin
    • RasterizedMap.xmax
    • RasterizedMap.ymax
    • RasterizedMap.map
    • RasterizedMap.xmin
    • RasterizedMap.ymin
    • RasterizedMap.xmax
    • RasterizedMap.ymax
    • RasterizedMap.__init__()
  • EmptyDatasetError
  • BaseDataset
    • BaseDataset.__init__()
    • BaseDataset.load_data()
    • BaseDataset.split_samples()
    • BaseDataset.collate_fn()
    • BaseDataset.resample_dataframe()
    • BaseDataset.normalize_xy()
    • BaseDataset.save_cache()
    • BaseDataset.load_cache()
• src.dataset.eth_dataset module
  • ETHDataset
    • ETHDataset.raw_fps
    • ETHDataset.load_data()
    • ETHDataset.load_data_batch()
• src.dataset.gc_dataset module
  • GCDataset
    • GCDataset.raw_fps
    • GCDataset.load_data()
    • GCDataset.get_homography_mat()
    • GCDataset.get_abnormal()
• src.dataset.orca_dataset module
  • ORCADataset
    • ORCADataset.raw_fps
    • ORCADataset.load_data()
    • ORCADataset.load_data_batch()
• src.dataset.sdd_dataset module
  • SDDDataset
    • SDDDataset.raw_fps
    • SDDDataset.load_data()
    • SDDDataset.load_data_batch()
    • SDDDataset.get_homography_mat()
    • SDDDataset.clean_trajectory()
• src.dataset.synthetic_dataset module
• src.dataset.ucy_dataset module
  • UCYDataset
    • UCYDataset.raw_fps
    • UCYDataset.load_data()
    • UCYDataset.load_data_batch()
    • UCYDataset.get_homography_mat()
• src.dataset.waymo_dataset module
  • WayMoDataset
    • WayMoDataset.raw_fps
    • WayMoDataset.load_data()
    • WayMoDataset.load_data_batch()
    • WayMoDataset.filter_vehicle_trajectories()
    • WayMoDataset.filter_short_trajectories()