rknn推理yolov5 的cmake
时间: 2023-08-22 21:11:55 浏览: 48
引用\[1\]:根据引用\[1\]中提到的信息,NCNN对于4维及以下的张量有较好的支持,而5维及以上的张量无法转换到ncnn。因此,在进行rknn推理yolov5的cmake时,需要注意张量的维度是否符合NCNN的要求。
引用\[2\]:关于TNN模型文件的下载地址,可以从Baidu Drive或者仓库中获取。根据引用\[2\]中提供的信息,可以下载不同大小的TNN模型文件,具体的文件命名和大小可以参考引用\[2\]中的列表。
引用\[3\]:在进行yolov5的cmake时,可以考虑不导出anchors和anchor_grid,这样可以减小模型文件的大小。根据引用\[3\]中的描述,修改后的模型文件大小为6.5Mb。
综上所述,进行rknn推理yolov5的cmake时,需要注意张量的维度是否符合NCNN的要求,并可以根据需要选择合适大小的TNN模型文件。此外,可以考虑不导出anchors和anchor_grid来减小模型文件的大小。具体的C++前后处理以及NMS的实现可以参考仓库的源码。
#### 引用[.reference_title]
- *1* *2* *3* [工程详细记录:超准确人脸检测(带关键点)YOLO5Face C++](https://blog.csdn.net/qq_29462849/article/details/122738829)[target="_blank" data-report-click={"spm":"1018.2226.3001.9630","extra":{"utm_source":"vip_chatgpt_common_search_pc_result","utm_medium":"distribute.pc_search_result.none-task-cask-2~all~insert_cask~default-1-null.142^v91^insert_down1,239^v3^insert_chatgpt"}} ] [.reference_item]
[ .reference_list ]