OpenCV + Tensorflow + PB

- Tensorflow로 저장한 pb파일 opencv에서 load해서 파라미터 찍어봄

코드

dnn::Net net = readNetFromTensorflow(modelFile); std::vector<String> alllayer = net.getLayerNames(); Mat inputBlob = blobFromImage(img, 1.0f, Size(224, 224), Scalar(), true, false); //Convert Mat to batch of images cv::Mat src =(Mat_<uchar>(3,3) << 1, 2, 3, 4, 5, 6, 7, 8, 9); cv::Mat bsrc = blobFromImage(src, 1.0f, Size(3, 3), Scalar(), true, false); cout<<src; net.setInput(bsrc, "I"); //set the network input for(int k = 0; k< alllayer.size(); k++) { String layername = alllayer[k].c_str(); int id = net.getLayerId(layername) ; printf("%d = %s, id num = %d \n", k, layername.c_str() , id); cv::dnn::Net::LayerId lid(layername); Ptr<Layer> cul = net.getLayer( lid ); std::vector<Mat> blob = cul->blobs; for( int k = 0; k < blob.size(); k++) { cout<<blob[k].size<<endl; cout<<blob[k]<<endl; } }

- 결과