dnn/NGraph: added nullptr checks

pull/17511/head
Maksim Shabunin 5 years ago
parent 781fbde449
commit 9096b1c768
  1. 2
      modules/dnn/src/dnn.cpp

@ -2023,6 +2023,7 @@ struct Net::Impl : public detail::NetImplBase
Ptr<BackendNode> inpNode = inpLd.backendNodes[preferableBackend]; Ptr<BackendNode> inpNode = inpLd.backendNodes[preferableBackend];
if (!inpNode.empty()) { if (!inpNode.empty()) {
Ptr<InfEngineNgraphNode> ieNode = inpNode.dynamicCast<InfEngineNgraphNode>(); Ptr<InfEngineNgraphNode> ieNode = inpNode.dynamicCast<InfEngineNgraphNode>();
CV_Assert(!ieNode.empty());
ieNode->net->setUnconnectedNodes(ieNode); ieNode->net->setUnconnectedNodes(ieNode);
} }
} }
@ -2067,6 +2068,7 @@ struct Net::Impl : public detail::NetImplBase
int cons_inp = cons->oid; int cons_inp = cons->oid;
Ptr<NgraphBackendWrapper> inpWrapper = inpLd.outputBlobsWrappers[cons_inp]. Ptr<NgraphBackendWrapper> inpWrapper = inpLd.outputBlobsWrappers[cons_inp].
dynamicCast<NgraphBackendWrapper>(); dynamicCast<NgraphBackendWrapper>();
CV_Assert(!inpWrapper.empty());
auto iter = std::find(inputNames.begin(), inputNames.end(), auto iter = std::find(inputNames.begin(), inputNames.end(),
inpWrapper->dataPtr->getName()); inpWrapper->dataPtr->getName());
if (iter == inputNames.end()) { if (iter == inputNames.end()) {

Loading…
Cancel
Save