This project contains early effort on converting mmdetection models into TVM Relay IR, which then can be deployed to different hardware, e.g. Nvidia Jetson.
We will mainly focus on two aspects to bring mmdetection models into production:
- Contribute to mmdetection to make all the models traceable. Currently, mmdetection is coded in a very Pythonic way that's hard to trace.
- Contribute to TVM for any missing operators / fix bugs for pytorch frontend. We also target at optimizing deployment performance of mmdetection models using Auto TVM.
As first step, we will focus on a few set of commonly used object detection models, as listed below:
- SSD
- Faster R-CNN
- Cascade R-CNN
- RetinaNet
- Set up a new conda environment:
conda create -n tvm python=3.7 - Install TVM from source
- Install mmdetection
- Hardware (optional)
VGG 16 backbone can be downloaded from here