├── CMakeLists.txt ├── README.md ├── cmake └── TritonRknnBackendConfig.cmake.in ├── doc ├── local_build_from_docker.md └── local_build_from_source.md ├── examples ├── ensemble_mobilenet │ └── test_ensemble_mobilenet.py ├── models │ ├── ensemble_mobilenet │ │ ├── 1 │ │ │ └── null.txt │ │ └── config.pbtxt │ ├── preprocess │ │ ├── 1 │ │ │ └── model.py │ │ └── config.pbtxt │ ├── single_input │ │ ├── 1 │ │ │ └── model.rknn │ │ └── config.pbtxt │ └── yolov5 │ │ ├── 1 │ │ └── model.rknn │ │ └── config.pbtxt ├── single_input │ ├── dog_224x224.jpg │ ├── labels.txt │ └── test_single_input.py └── yolov5 │ ├── bus.jpg │ ├── test_yolov5.py │ └── yolov5_result.jpg ├── rknpu2 ├── include │ ├── rknn_api.h │ └── rknn_matmul_api.h └── lib │ ├── librknn_api.so │ └── librknnrt.so └── src ├── libtriton_rknn.ldscript ├── rknn.cc ├── rknn_utils.cc └── rknn_utils.h /CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/CMakeLists.txt -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/README.md -------------------------------------------------------------------------------- /cmake/TritonRknnBackendConfig.cmake.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/cmake/TritonRknnBackendConfig.cmake.in -------------------------------------------------------------------------------- /doc/local_build_from_docker.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/doc/local_build_from_docker.md -------------------------------------------------------------------------------- /doc/local_build_from_source.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/doc/local_build_from_source.md -------------------------------------------------------------------------------- /examples/ensemble_mobilenet/test_ensemble_mobilenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/ensemble_mobilenet/test_ensemble_mobilenet.py -------------------------------------------------------------------------------- /examples/models/ensemble_mobilenet/1/null.txt: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /examples/models/ensemble_mobilenet/config.pbtxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/ensemble_mobilenet/config.pbtxt -------------------------------------------------------------------------------- /examples/models/preprocess/1/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/preprocess/1/model.py -------------------------------------------------------------------------------- /examples/models/preprocess/config.pbtxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/preprocess/config.pbtxt -------------------------------------------------------------------------------- /examples/models/single_input/1/model.rknn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/single_input/1/model.rknn -------------------------------------------------------------------------------- /examples/models/single_input/config.pbtxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/single_input/config.pbtxt -------------------------------------------------------------------------------- /examples/models/yolov5/1/model.rknn: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/yolov5/1/model.rknn -------------------------------------------------------------------------------- /examples/models/yolov5/config.pbtxt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/models/yolov5/config.pbtxt -------------------------------------------------------------------------------- /examples/single_input/dog_224x224.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/single_input/dog_224x224.jpg -------------------------------------------------------------------------------- /examples/single_input/labels.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/single_input/labels.txt -------------------------------------------------------------------------------- /examples/single_input/test_single_input.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/single_input/test_single_input.py -------------------------------------------------------------------------------- /examples/yolov5/bus.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/yolov5/bus.jpg -------------------------------------------------------------------------------- /examples/yolov5/test_yolov5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/yolov5/test_yolov5.py -------------------------------------------------------------------------------- /examples/yolov5/yolov5_result.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/examples/yolov5/yolov5_result.jpg -------------------------------------------------------------------------------- /rknpu2/include/rknn_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/rknpu2/include/rknn_api.h -------------------------------------------------------------------------------- /rknpu2/include/rknn_matmul_api.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/rknpu2/include/rknn_matmul_api.h -------------------------------------------------------------------------------- /rknpu2/lib/librknn_api.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/rknpu2/lib/librknn_api.so -------------------------------------------------------------------------------- /rknpu2/lib/librknnrt.so: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/rknpu2/lib/librknnrt.so -------------------------------------------------------------------------------- /src/libtriton_rknn.ldscript: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/src/libtriton_rknn.ldscript -------------------------------------------------------------------------------- /src/rknn.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/src/rknn.cc -------------------------------------------------------------------------------- /src/rknn_utils.cc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/src/rknn_utils.cc -------------------------------------------------------------------------------- /src/rknn_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zjd1988/rknn_backend/HEAD/src/rknn_utils.h --------------------------------------------------------------------------------