search

dlunion / tensorRTIntegrate

TensorRT ONNX Plugin、Inference、Compile
465 stars 114 forks source link

根据dladcn_export_onnx.py生成的onnx转成trtmodel一个都检测不到 #18

Open DavidLeeDVD opened 4 years ago

DavidLeeDVD commented 4 years ago

根据https://github.com/dlunion/tensorRTIntegrate/tree/master/dcn_onnx将官方下载的原始pth转成onnx然后再转成trtmodel一个都检测不到，已经按照文案中所说将dcn_v2.py和pose_dla_dcn.py替换原文件，生成出来的onnx文件也用netron打开看过网络结构都是一样的，唯二的不同点是输入是input.1，标准的是0；生成的用的是pytorch1.2，标准的是pytorch1.1。不明白为什么一个都检测不到？

hopef commented 4 years ago

你可以发邮件到512690069@qq.com，或者添加QQ群737800788。我帮你确定问题