search

CaoWGG / TensorRT-YOLOv4

tensorrt5, yolov4, yolov3,yolov3-tniy,yolov3-tniy-prn
264 stars 76 forks source link

关于onnx生成engine的问题 #24

Open qualia0000 opened 4 years ago

qualia0000 commented 4 years ago

大佬您好，我在生成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生成产生什么影响？ 感谢大佬！

CaoWGG commented 4 years ago

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

qualia0000 commented 4 years ago

@CaoWGG 好的，感谢!