TensorRT-YOLOv4 icon indicating copy to clipboard operation
TensorRT-YOLOv4 copied to clipboard

关于onnx生成engine的问题

Open qualia0000 opened this issue 5 years ago • 2 comments

大佬您好,我在生成engine的时候注意到opset version的问题。在ubuntu和您相同的环境下生成engine时opset version是12,我在windows下生成engine时则是0。在代码中我注意到: ModelImporter.cpp line 239: int64_t opset_version = (onnx_model.opset_import().size() ? onnx_model.opset_import(0).version() : 0); 根据这一句,我是否可以确定此时onnx_model.opset_import().size() 为0?若确实如此,这将对engine生成产生什么影响? 感谢大佬!

qualia0000 avatar Jun 10 '20 10:06 qualia0000

@qualia0000 抱歉,没在windows下转过模型,不清楚啊。 不同的opset_version,某些op的表示不同,比如resize这个op

CaoWGG avatar Jun 18 '20 02:06 CaoWGG

@CaoWGG 好的,感谢!

qualia0000 avatar Jun 18 '20 02:06 qualia0000