TRT8 works with almost every nerual network now!!!!(including demo3d)

2021-10-28 23:35:37 +05:30
parent 8c36dd0431
commit c5e66c6bf6
37 changed files with 1042 additions and 700 deletions
@@ -290,6 +290,6 @@ int main()
    
    cv::Mat viz = vizLayer2Mat(&net, net.num_layers-1);
    cv::imwrite("test.png", viz);
-
+    netRT.destroy();
    return ret_cudnn | ret_tensorrt | ret_cudnn_tensorrt;
 }