基于C++和OpenCV实现基于暗通道先验的图像去雾算法+源代码+文档说明(高分课程设计)

#include "ImageDehazing.h" #include "readImg.h" #include "glog/logging.h" #include "gflags/gflags.h" #if _DEBUG #pragma comment(lib,"gflags_static_d.lib") #pragma comment(lib,"gflags_nothreads_static_d.lib") #else #pragma comment(lib,"gflags_static.lib") #pragma comment(lib,"gflags_nothreads_static.lib") #endif #pragma comment(lib,"shlwapi.lib") //视频去雾; void VideoDehaze(string& videofile) { CImageDehazing dehaze; VideoCapture cap; cap.open(videofile); if (!cap.isOpened()) { return; } Mat frame; while (cap.read(frame)) { Mat dehazed_img; CStatisticalTime statisticaltime; statisticaltime.startTime(); dehaze.process(frame, dehazed_img); cout << statisticaltime.endTime() << endl; imshow("original img", frame); imshow("dehaze img", dehazed_img); if(27 == waitKey(15)) break; } destroyAllWindows(); } //图像去雾; void ImageDehaze(string& imagepath) { CImageDehazing dehaze; Mat img = imread(imagepath); //dehaze.resize_img(img, img); Mat dehazed_img; dehaze.process(img, dehazed_img); imshow("original img", img); imshow("dehaze img", dehazed_img); //imwrite("../non_local_dehazing_code/non_local_dehazing/images/originalimg.jpg", img); imwrite("../../images/dehazed_he_vs.jpg", dehazed_img); waitKey(0); } //文件夹图像去雾; void FolderDehaze_image(string& folderpath) { CImageDehazing dehaze; readImg rdImage; const string feature = "*"; vector<string> filenames; rdImage.readImage(folderpath, feature, filenames); auto it = filenames.begin(); while (it != filenames.end()) { Mat img = imread(*it); dehaze.resize_img(img, img); Mat dehazed_img; CStatisticalTime statisticaltime; statisticaltime.startTime(); dehaze.process(img, dehazed_img); cout << statisticaltime.endTime() << endl; //imwrite("Org_images/original.jpg", img); //imwrite("Org_images/dehazed.jpg", dehazed_img); cout << *it << endl; imshow("original", img); imshow("dehaze", dehazed_img); waitKey(0); it++; } } DEFINE_int32(user_choice, 0, "0:nothing\n 1:video/n 2:path\n 3:image\n"); DEFINE_string(video_name, "test.avi", "video file name"); DEFINE_string(image_path, "test", "image file path"); DEFINE_string(image_name, "test.jpg", "image file name"); int main(int argc,char* argv[]) { google::InitGoogleLogging(argv[0]); gflags::ParseCommandLineFlags(&argc, &argv, true); switch (FLAGS_user_choice) { case 1: VideoDehaze(FLAGS_video_name); break; case 2: FolderDehaze_image(FLAGS_image_path); break; case 3: ImageDehaze(FLAGS_image_name); break; default: break; } }