MTrajRec代码阅读笔记

1.MTrajRec算法解决的问题

MTrajRec算法主要是将gps轨迹从低采样数据中恢复/预测出一个高采样的轨迹数据，并能够与地图上面的道路相互对应，它是一个端到端的算法，无需二次匹配。

2.数据准备

2.1下载地图数据

从https://download.geofabrik.de/下载数据https://download.geofabrik.de/asia/china-latest.osm.pbf
从Geofabrik下载格式化的OSM数据。

wget https://download.geofabrik.de/asia/china-latest.osm.pbf

2.2下载GPS轨迹数据

网页https://www.microsoft.com/en-us/research/publication/t-drive-trajectory-data-sample/

下载链接：https://1drv.ms/u/s!AsWQUIUFkRXPgsN6OCa9OB-qziGfhA

2.3裁剪地图数据

代码参考https://github.com/sjruan/osm2rn
根据感兴趣的区域裁剪数据。

python osm_clip.py --input_path china-latest.osm.pbf --output_path interest_region.osm.pbf --min_lat 39.8451 --min_lng 116.2810 --max_lat 39.9890 --max_lng 116.4684

根据裁剪的数据构建和存储道路网络。

python osm_to_rn.py --input_path interest_region.osm.pbf --output_path interest_region

得到几个文件，格式如下：

2.4清理轨迹数据

python main.py --phase clean --tdrive_root_dir ./data/taxi_log_2008_by_id/ --clean_traj_dir ./data/tdrive_clean/ 

清理前：

清理后：

2.4将轨迹数据和地图数据转换成训练所需数据格式

python main.py --phase mm --clean_traj_dir ./data/tdrive_clean/ --rn_path ./data/interest_region/ --mm_traj_dir ./data/tdrive_mm/

clean_traj_dir :清理的轨迹数据路径
interest_region:地图数据路径
tdrive_mm:转换后的数据路径

2.5 训练

项目地址 MTrajRec

训练时将tdrive_mm数据放在项目 data/raw_trajectory/ 文件夹下

运行multi_main.py后将自动划分训练集、验证集、测试集到 data/model_data 目录下