yolov3 c++ tensorflow inference

    cv_data.image = cv::imread(image_path, -1);
    cv_data.image_nms = cv_data.image.clone();

    data_config.scale = std::min(1.0*data_config.iw/data_config.w, 1.0*data_config.ih/data_config.h);

    data_config.nw = int(data_config.scale * data_config.w);
    data_config.nh = int(data_config.scale * data_config.h);

    cv::Mat image_resized;
    cv::resize(cv_data.image, image_resized, cv::Size(data_config.nw, data_config.nh));

    cv::Mat image_paded(data_config.ih, data_config.iw, CV_32FC3, cv::Scalar(128, 128, 128));
    data_config.dw = (int)(data_config.iw - data_config.nw)/2.0;
    data_config.dh = (int)(data_config.ih - data_config.nh)/2.0;
    
    // std::cout << dw << " " << dh << " " << dw+nw << " " << dh+nh << std::endl;
    cv::Rect roi_rect = cv::Rect(data_config.dw, data_config.dh, data_config.nw, data_config.nh);
    image_resized.copyTo(image

你可能感兴趣的:(TensorFlow)