Releases: ceccocats/tkDNN
Releases · ceccocats/tkDNN
version 0.6
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
version 0.5
works with:
CUDA 10.0
CUDNN 7.603
TENSORRT 6.01
OPENCV 4.1
NVIDIA jetpack 4.3
New features:
- batch inference
- Darknet cfg parser
- yolo4
- centernet
- minor improvements
Supported networks
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-mobilenetv2ssdversion 0.4
works with:
CUDA 10.0
CUDNN 7.603
TENSORRT 6.01
OPENCV 4.1
NVIDIA jetpack 4.3
