D2Go is a production ready software system from FacebookResearch, which supports end-to-end model training and deployment for mobile platforms.
- It is a deep learning toolkit powered by PyTorch and Detectron2.
- State-of-the-art efficient backbone networks for mobile devices.
- End-to-end model training, quantization and deployment pipeline.
- Easy export to TorchScript format for deployment.
Install PyTorch Nightly (use CUDA 10.2 as example, see details at PyTorch Website):
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch-nightly
Install Detectron2 (other installation options at Detectron2):
python -m pip install 'git+https://github.com/facebookresearch/detectron2.git'
python -m pip install 'git+https://github.com/facebookresearch/mobile-vision.git'
git clone https://github.com/facebookresearch/d2go cd d2go & python -m pip install .
D2Go is released under the Apache 2.0 license.