源码根目录 (Source Root)

Subpackages

• 数据加载与处理 (Dataset)
  • Submodules
    • src.dataset.base_dataset module
      • RasterizedMap
      • EmptyDatasetError
      • BaseDataset
    • src.dataset.eth_dataset module
      • ETHDataset
    • src.dataset.gc_dataset module
      • GCDataset
    • src.dataset.orca_dataset module
      • ORCADataset
    • src.dataset.sdd_dataset module
      • SDDDataset
    • src.dataset.synthetic_dataset module
    • src.dataset.ucy_dataset module
      • UCYDataset
    • src.dataset.waymo_dataset module
      • WayMoDataset
• 扩散生成算法 (Diffusion)
  • Submodules
    • src.diffusion.ddim module
      • DDIM
    • src.diffusion.ddpm module
      • DDPM
• 核心预测模型 (Model)
  • Submodules
    • src.model.fourier_positional_encoding module
      • FourierPositionalEncoding
    • src.model.mean_pooling_lstm module
      • MeanPoolingLSTM
    • src.model.model module
      • Model
    • src.model.multiscale_cnn module
      • MultiScaleCNN
    • src.model.nan_embedding module
      • NanEmbedding
    • src.model.new_model module
      • NewModel
    • src.model.permuted module
      • Permuted
    • src.model.relative_model module
      • RelativeModel
    • src.model.residual module
      • Residual
    • src.model.sinusoidal_embedding module
      • SinusoidalEmbedding
• src.tasks package
  • Submodules
    • src.tasks.guidance module
      • guidance()
    • src.tasks.sfm module
    • src.tasks.simulate module
      • SimulateState
      • init_simulation()
      • simulate_one_step()
    • src.tasks.test_once module
      • get_xy_error()
      • get_collision_rate()
      • get_collision_rate2()
      • test_once()
    • src.tasks.train_once module
      • train_once()
    • src.tasks.visualize module
      • visualize()
• 通用工具组件 (Utils)
  • Submodules
    • src.utils.auto_gpu module
      • AutoGPU
    • src.utils.calc_xy_error module
      • calc_xy_error()
    • src.utils.crop_axes_region module
      • crop_axes_region()
    • src.utils.extract_patches module
      • extract_patches_numpy()
      • extract_patches_torch()
    • src.utils.fix_parser module
      • add_minus_flags()
      • add_negation_flags()
    • src.utils.get_force_map module
      • get_force_map()
    • src.utils.homography module
      • calc_homography_mat()
      • affine_transformation()
      • image_to_world()
    • src.utils.json_compatible module
      • json_compatible()
    • src.utils.logger module
      • init_logger()
    • src.utils.plot module
      • get_fig()
      • plot_resilience()
      • EqualizeNormalize
      • plotOD()
      • clear_svg()
      • load_font()
      • use_chinese_font()
      • merge_axes()
    • src.utils.seed module
      • seed_all()
    • src.utils.tag2ansi module
      • tag2ansi()
    • src.utils.timer module
      • Timer
      • AbsTimer
      • NamedTimer
      • ParallelTimer
    • src.utils.use_npu module
      • npu_attention_fallback()
      • recover_npu_attention()
      • npu_attention_fallback_context()