1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283 |
- #include <opencv2/opencv.hpp>
- #include <iostream>
- #pragma comment(lib, "opencv_world4100.lib")
- using namespace cv;
- using namespace std;
- void removeShadow(const Mat& src, Mat& dst) {
-
- Mat hsv;
- cvtColor(src, hsv, COLOR_BGR2HSV);
-
- vector<Mat> hsvChannels;
- split(hsv, hsvChannels);
-
- Mat vChannel = hsvChannels[2];
- Mat element = getStructuringElement(MORPH_RECT, Size(3, 3));
- int iteration = 9;
-
- Mat dilateMat;
- morphologyEx(vChannel, dilateMat, MORPH_DILATE, element, Point(-1, -1), iteration);
- Mat erodeMat;
- morphologyEx(dilateMat, erodeMat, MORPH_ERODE, element, Point(-1, -1), iteration);
-
- Mat calcMat = ~(erodeMat - vChannel);
- normalize(calcMat, calcMat, 0, 255, NORM_MINMAX);
-
- hsvChannels[2] = calcMat;
- merge(hsvChannels, hsv);
-
- cvtColor(hsv, dst, COLOR_HSV2BGR);
- }
- int main(int argc, char** argv) {
-
- Mat src = imread("test.webp");
- if (src.empty()) {
- cerr << "无法加载图像\n";
- return -1;
- }
- Mat result;
- removeShadow(src, result);
-
- double alpha = 1.1;
- int beta = 5;
- convertScaleAbs(result, result, alpha, beta);
-
- Size srcSize = src.size();
-
- int displayHeight = 1080;
- int displayWidth = srcSize.width * displayHeight / srcSize.height;
-
- Mat resizedSrc, resizedResult;
- resize(src, resizedSrc, Size(displayWidth, displayHeight));
- resize(result, resizedResult, Size(displayWidth, displayHeight));
-
- Mat combined;
- hconcat(resizedSrc, resizedResult, combined);
-
- imshow("原图和去除阴影后的图像对比", combined);
- waitKey(0);
-
- imwrite("output.jpg", result);
- return 0;
- }
|