C++ format input

medium_yolov5_frame_input.cpp

cv::Mat format_yolov5(const cv::Mat &source) {
  
    // put the image in a square big enough
    int col = source.cols;
    int row = source.rows;
    int _max = MAX(col, row);
    cv::Mat resized = cv::Mat::zeros(_max, _max, CV_8UC3);
    source.copyTo(resized(cv::Rect(0, 0, col, row)));
    
    // resize to 640x640, normalize to [0,1[ and swap Red and Blue channels
    cv::Mat result;
    cv::dnn::blobFromImage(source, result, 1./255., cv::Size(INPUT_WIDTH, INPUT_HEIGHT), cv::Scalar(), true, false);
  
    return result;
}