基于TensorRT加速深度學(xué)習(xí)圖像識別的方法、裝置及介質(zhì)

基本信息

申請?zhí)?/td> CN202011206759.7 申請日 -
公開（公告）號 CN112364744A 公開（公告）日 2021-02-12
申請公布號 CN112364744A 申請公布日 2021-02-12
分類號 G06K9/00(2006.01)I; 分類 計算；推算；計數(shù);
發(fā)明人 李偉;李翔;汪凡;車志宏;譚必文 申請（專利權(quán)）人 珠海市卓軒科技有限公司
代理機構(gòu) 廣州嘉權(quán)專利商標事務(wù)所有限公司 代理人 葉琦煒
地址 519000廣東省珠海市橫琴新區(qū)寶華路6號105室
法律狀態(tài) -

摘要

摘要 本發(fā)明涉及一種基于TensorRT加速深度學(xué)習(xí)圖像識別的方法、裝置及介質(zhì)的技術(shù)方案，包括：配置用于運行TensorRT的數(shù)據(jù)、模型及環(huán)境；將模型轉(zhuǎn)換為ONNX格式；通過TensorRT對ONNX格式的模型分別生成多計算模式運算的TensorRT引擎文件；輸入用于檢測的多張人臉照片，通過MTCNN網(wǎng)絡(luò)對人臉照片進行人臉檢測，輸出檢測列表；將檢測列表中的人臉照片通過包括有TensorRT的多種加速深度學(xué)習(xí)圖像識別的推理計算，分別推理計算人臉照片的向量、時間及距離，選取加速效果最優(yōu)的TensorRT加速計算模式。本發(fā)明的有益效果為：保持原有精確率幾乎無損的情況下，同時也提升了GPU服務(wù)器的資源利用率，降低了業(yè)務(wù)升級和維護方面的人力及物力。??