-
-
strong
能不能把你们的训练模型solov2_lite_coco_704.pth和相关配置文件分享一下我在我电脑上看能不能得到你们分享的solov2_lite_coco_704.onnx
-
strong
@刘看山
ONNX IR version: 0.0.6
Opset version: 11
Producer name: pytorch
Producer version: 1.7
Domain:
Model version: 0
Doc string:[08/12/2021-16:53:18] [W] [TRT] onnx2trt_utils.cpp:220: Your ONNX model has been generated with INT64 weights, while TensorRT does not natively support INT64. Attempting to cast down to INT32.
[08/12/2021-16:53:18] [W] [TRT] Tensor DataType is determined at build time for tensors not marked as input or output.
ERROR: builtin_op_importers.cpp:1602 In function importIf:
[8] Assertion failed: cond.is_weights() && cond.weights().count() == 1 && "If condition must be a initializer!"
[08/12/2021-16:53:18] [E] Parsing File Failed
[08/12/2021-16:53:18] [E] Init Session Failed (attempt from onnx file)! -
strong
已经完成步骤:
1. 工具链Solov2_trt的demo跑通,可以把分享的模型solov2_lite_coco_704.onnx转为solov2_lite_coco_704.trt并完成推理
2. Solov2_d2中完成自己数据集的训练,可以在demo.py中完成推理
3. 利用solov2_d2工程中demo/export.py可以将自己的训练模型导出为solov2.onnx
问题
在solov2_trt中将solov2.onnx转换为solov.trt时报错,错误如下:
安装环境:
pytorch1.7.1
tensorrt7.2.3.4
cuda11.1