SSD 模型的C++推理部署
Created by: lianDaniel
在C++推理部署中,配置中的FEEDS_SIZE表示 # 输入的tensor数量,其中3是faster rcnn
,2是yolo,并且指出了一般的模型这个值为1,但是在文件preprocessor/preprocessor_detection.cpp,在feeds_size为2的情况,却没有图像读取代码。
cv::Mat im1 = cv::imread(fname, -1); cv::Mat im; if (_config->_feeds_size == 3) { // faster rcnn im1.convertTo(im, CV_32FC3, 1/255.0); } else if (_config->_feeds_size == 2) { // yolo v3 im = im1; } if (im.data == nullptr || im.empty()) { #ifdef _WIN32 std::cerr << "Failed to open image: " << fname << std::endl; #else LOG(ERROR) << "Failed to open image: " << fname; #endif return false; }
不知道是否是在别的地方有所设置?或者说是不是当前的C++推理代码并没有考虑SSD模型。