• S
    strong

    @刘看山
    用onnx-simplifier简化后问题解决,谢谢

    发布在 社区求助区(SOS!!) 阅读更多
  • S
    strong

    @刘看山 @jinfagang

    能不能把你们的训练模型solov2_lite_coco_704.pth和相关配置文件分享一下我在我电脑上看能不能得到你们分享的solov2_lite_coco_704.onnx

    发布在 社区求助区(SOS!!) 阅读更多
  • S
    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)!

    发布在 社区求助区(SOS!!) 阅读更多
  • S
    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时报错,错误如下:
    25e7c2e9-eca3-4d6f-9e56-1fa09f91da3c-image.png
    安装环境:
    pytorch1.7.1
    tensorrt7.2.3.4
    cuda11.1

    发布在 社区求助区(SOS!!) 阅读更多

与 神力AI社区 的连接断开,我们正在尝试重连,请耐心等待