舒服花瓣

文章
7
资源
0
加入时间
2年10月21天

torch版ResNet50(带有多输出)转c++ tensorrt

第一个最简单的lenet示例请参考这篇文章一.torch阶段测试图片:torch代码:# coding:utf-8import torchfrom torch import nnfrom torch.nn import functional as Fimport torchvisionimport osimport structimport timeimport cv2import numpy as npdef main(): print('cuda d.