Works with:
- CUDA 11.0
- CUDNN 8.0.4
- TENSORRT 7.2.0
- OPENCV 4.5.2
NVIDIA jetpack 4.4
New features:
- ShelfNet (semantic segmentation)
- CenterNet3D (3D object detection)
- CenterTrack2D and 3D (2D/3D object detection and tracking)
- Scaled Yolov4 (yolov4-csp)
- Windows support
Supported networks
shelfnet_berkeley shelfnet dla34_ctrack dla34_cnet3d yolo4-csp yolo4x yolo4 yolo4_berkeley yolo3 yolo3_berkeley yolo3_coco4 yolo3_flir yolo3_512 yolo3tiny yolo3tiny_512 yolo2 yolo2_voc csresnext50-panet-spp resnet101_cnet dla34_cnet mobilenetv2ssd mobilenetv2ssd512 bdd-mobilenetv2ssdD