Skip to content

opencv dnn can't load pytorch FasterRCNN exported to onnx detection model #21634

@chenliang2014

Description

@chenliang2014

Environment:

  • OpenCV => 4.5.5
  • Operating System / Platform => Windows10 64 Bit
  • pytorch 1.7
  • python 3.7

export pytorch model to onnx code:
`
onnx_model_path = "./fasterrcnn_resnet50_fpn.onnx"
model = torchvision.models.detection.fasterrcnn_resnet50_fpn(pretrained=True)
device = torch.device('cpu')

model.to(device)
dummy_input = torch.randn(1, 3, 608, 608).to(device)
model.eval()
torch.onnx.export(model, dummy_input, onnx_model_path, verbose=False, opset_version=11)
`

In opencv load onnx model code:
net = cv2.dnn.readNetFromONNX(onnx_model_path)

report error:

net = cv2.dnn.readNetFromONNX(onnx_model_path)
cv2.error: OpenCV(4.5.5) D:\a\opencv-python\opencv-python\opencv\modules\dnn\src\graph_simplifier.cpp:76: error: (-212:Parsing error) Input node with name 1625 not found in function 'cv::dnn::Subgraph::getInputNodeId'

Metadata

Metadata

Assignees

Labels

Type

No type

Projects

No projects

Milestone

Relationships

None yet

Development

No branches or pull requests

Issue actions