文件名称：bidireaction-trajectory-prediction:双向轨迹预测（BiTraP）的代码
文件大小：476KB
文件格式：ZIP
更新时间：2024-06-12 08:30:14
Python
BiTraP：具有多模式目标估计的双向行人轨迹预测 余瑶，埃拉·阿特金斯（Ella Atkins），马修·约翰逊·罗伯森（Matthew Johnson-Roberson），拉姆·维苏德万（Ram Vasudevan）和小晓杜 此回购包含我们论文的代码： 。 我们的BiTraP-NP网络架构： 我们的BiTraP-GMM解码器架构： 安装 依存关系 我们的代码是使用python和pytorch实现的，并在具有Intel Xeon 2.10GHz CPU，NVIDIA TITAN X GPU和128 GB内存的台式计算机上进行了测试。 NVIDIA驱动程序> = 418 Python> = 3.6 pytorch == 1.4.1，具有GPU支持（CUDA 10.1＆cuDNN 7） 运行以下命令以将bitrap路径添加到PYTHONPATH cd bidireaction-t
【文件预览】：
bidireaction-trajectory-prediction-main
----.gitignore(146B)
----requirements.txt(105B)
----bitrap()
--------modeling()
--------layers()
--------structures()
--------engine()
--------utils()
----datasets()
--------preprocessing.py(12KB)
--------ETH_UCY.py(7KB)
--------environment()
--------JAAD.py(10KB)
--------PIE.py(11KB)
--------transforms.py(4KB)
--------__init__.py(3KB)
--------JAAD_origin.py(59KB)
--------PIE_origin.py(63KB)
----.gitmodules(84B)
----tools()
--------__init__.py(0B)
--------test.py(2KB)
--------train.py(5KB)
----README.md(3KB)
----configs()
--------bitrap_np_PIE.yml(1KB)
--------bitrap_np_ETH.yml(915B)
--------__init__.py(32B)
--------bitrap_gmm_ETH.yml(1KB)
--------bitrap_gmm_JAAD.yml(1KB)
--------ETH_UCY.json(2KB)
--------bitrap_np_JAAD.yml(1KB)
--------defaults.py(2KB)
--------bitrap_gmm_PIE.yml(1KB)
----figures()
--------bitrap_np.png(243KB)
--------bitrap_gmm.png(127KB)
----docker()
--------Dockerfile(350B)