Android Demon of mobilev2-yolo5s and retinaface。深度学习模型工程化优化的最后一步。
模型端轻量化,剪枝,蒸馏 ===>https://github.com/Syencil/mobile-yolov5-pruning-distillation
云端和嵌入式端模型转换 ===> https://github.com/Syencil/tensorRT
- 实现检测模型的demo,模型为mobilev2-yolo5s和retinaface
- 实现模型之间的一键转换,图像大小的一键转换(640 ===> 320)。返回的Toast时间为是从java代码请求C++开始到得到结果的总时间,而benchmark返回的fps仅为infer的时间
- 目前仅测试过华为P40 pro的CPU端
此android只是一个方便展示的前端界面。C++均为本人所写,JAVA大部分参考了其他git。具体优化方式和内容请移步mobilev2-yolo5s