Skip to content

bmedeir4/route-extraction-Deep-Learning-Model

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Geometry Translation

Feature Extraction

python feature_extraction.py ../data/conf_tdrive_sample.json

Training

python train.py --name t2rnet_tdrive_sample --dataroot ../data/tdrive_sample/learning --lam 0.2 --batch_size 8 --model t2rnet --display_id -1

Inference

python test.py --name t2rnet_tdrive_sample --dataroot ../data/tdrive_sample/learning --lam 0.2 --model t2rnet

Region Concatenation

python region_concatenation.py --tile_path ./results/t2rnet_tdrive_sample/test_latest/images/ --conf_path ../data/conf_tdrive_sample.json --results_path ../data/tdrive_sample/results/

Topology Construction

Graph Extraction

python main.py --phase 1 --conf_path ../data/conf_tdrive_sample.json --results_path ../data/tdrive_sample/results/

Link Generation

python main.py --phase 2 --conf_path ../data/conf_tdrive_sample.json --results_path ../data/tdrive_sample/results/

Map Refinement

Map Matching

python main.py --phase 3 --conf_path ../data/conf_tdrive_sample.json --results_path ../data/tdrive_sample/results/

Edge Pruning

python main.py --phase 4 --conf_path ../data/conf_tdrive_sample.json --results_path ../data/tdrive_sample/results/

Requirements

DeepMG uses the following dependencies with Python 3.6

  • gdal==2.3.2
  • opencv==3.3.1
  • rtree==0.8.3
  • networkx==2.3
  • scikit-image==0.16.2
  • pytorch==1.1.0
  • torchvision==0.3.0

Other packages can be easily installed using conda install, while the following scripts are recommended for gdal, opencv and pytorch.

conda install -c conda-forge gdal==2.3.2

conda install -c menpo opencv==3.3.1

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch -c defaults -c numba/label/dev

Note that gdal must be installed first, and a restart might be required after all installation.

Datasets

  • Trajectory Data

    • There are some open-source trajectory datasets, e.g., TDrive.
    • Data should be organized as text files. A text file can contain several trajectories (refer to sample.txt.template)
  • Road Network Data

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages