|
|
#ifndef QB_IMGPROCESS_H
|
|
|
#define QB_IMGPROCESS_H
|
|
|
|
|
|
#ifdef IMGPROCESS_EXPORTS
|
|
|
#define IMGPROCESS_API __declspec(dllexport)
|
|
|
#else
|
|
|
#define IMGPROCESS_API __declspec(dllimport)
|
|
|
#endif
|
|
|
|
|
|
// OpenCV
|
|
|
#include <opencv2\opencv.hpp>
|
|
|
using namespace cv;
|
|
|
|
|
|
// 情报数据通用结构
|
|
|
#include "QBStru.h"
|
|
|
|
|
|
|
|
|
// 1. 图像去雾控制 算法：王家星
|
|
|
|
|
|
//1.1 功能：图像去雾控制（暗通道先验）
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. A_MAX 控制去雾效果, A_MAX < 0 时，自适应计算(特别耗时)； 有效范围【0,255】
|
|
|
// 3. degree 控制去雾效果, 有效范围【0,1】
|
|
|
// 4. bSrcFirst 默认时，优先处理srcImg, srcImg-->dstImg
|
|
|
// 否则，优先处理dstImg，dstImg——>dstImg
|
|
|
|
|
|
//输出： 1. qbData 含去雾后图像的情报数据结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
//说明: 首先判断图像数据有效性标识位，无效时直接返回，不做任何处理;
|
|
|
IMGPROCESS_API bool DoDehazeCtrl(QBStru *qbData,
|
|
|
int A_MAX = -1,
|
|
|
double degree = 0.78,
|
|
|
bool bSrcFirst = true);
|
|
|
|
|
|
|
|
|
//1.2 功能：图像去雾控制（函数重载）
|
|
|
//输入： 1. src 输入图像结构体
|
|
|
// 2. A_MAX 控制去雾效果, A_MAX < 0 时，自适应计算(特别耗时)； 有效范围【0,255】
|
|
|
// 3. degree 控制去雾效果, 有效范围【0,1】
|
|
|
|
|
|
//输出： 1. dst 输出图像结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoDehazeCtrl(ImgStru* src,
|
|
|
ImgStru* dst,
|
|
|
int A_MAX = -1,
|
|
|
double degree = 0.78);
|
|
|
|
|
|
|
|
|
|
|
|
// 2. 图像亮度控制 算法：王家星
|
|
|
|
|
|
//2.1 功能：图像亮度控制
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. degree 控制亮度, 有效范围【0,20】，值越小越暗，越大越亮
|
|
|
// 3. bSrcFirst 默认时，优先处理srcImg, srcImg-->dstImg
|
|
|
// 否则，优先处理dstImg， dstImg——>dstImg
|
|
|
|
|
|
//输出： 1. qbData 含亮度变化后图像的情报数据结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
//说明: 首先判断图像数据有效性标识位，无效时直接返回，不做任何处理;
|
|
|
IMGPROCESS_API bool DoIntensityCtrl(QBStru *qbData,
|
|
|
int degree = 10,
|
|
|
bool bSrcFirst = true);
|
|
|
|
|
|
|
|
|
//2.2 功能：图像亮度控制（函数重载）
|
|
|
//输入： 1. src 输入图像结构体
|
|
|
// 2. degree 控制亮度,有效范围【0,20】，值越小越暗，越大越亮
|
|
|
|
|
|
//输出： 1. dst 输出图像结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoIntensityCtrl(ImgStru* src,
|
|
|
ImgStru* dst,
|
|
|
int degree = 10);
|
|
|
|
|
|
|
|
|
|
|
|
// 3. 图像对比度控制 算法：王家星
|
|
|
|
|
|
//3.1 功能：图像对比度控制
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. degree 控制对比度
|
|
|
// 3. bSrcFirst 默认时，优先处理srcImg, srcImg-->dstImg
|
|
|
// 否则，优先处理dstImg，dstImg——>dstImg
|
|
|
|
|
|
//输出： 1. qbData 含对比度变化后图像的情报数据结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
//说明: 首先判断图像数据有效性标识位，无效时直接返回，不做任何处理;
|
|
|
IMGPROCESS_API bool DoContrastCtrl(QBStru *qbData,
|
|
|
int degree = 10,
|
|
|
bool bSrcFirst = true);
|
|
|
|
|
|
//3.2 功能：图像对比度控制（函数重载）
|
|
|
//输入： 1. src 输入图像结构体
|
|
|
// 2. degree 对比度亮度,有效范围【0,20】，值越大对比度越强
|
|
|
|
|
|
//输出： 1. dst 输出图像结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoContrastCtrl(ImgStru* src,
|
|
|
ImgStru* dst,
|
|
|
int degree = 10);
|
|
|
|
|
|
|
|
|
// 4. 图像清晰度
|
|
|
|
|
|
//4.1 功能：图像清晰度控制
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. degree 控制清晰度
|
|
|
// 3. bSrcFirst 默认时，优先处理srcImg, srcImg-->dstImg
|
|
|
// 否则，优先处理dstImg，dstImg——>dstImg
|
|
|
|
|
|
//输出： 1. qbData 含清晰度变化后图像的情报数据结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
//说明: 首先判断图像数据有效性标识位，无效时直接返回，不做任何处理;
|
|
|
IMGPROCESS_API bool DoDefinitionCtrl(QBStru *qbData,
|
|
|
int degree = 10,
|
|
|
bool bSrcFirst = true);
|
|
|
|
|
|
|
|
|
//4.2 功能：图像清晰度控制（函数重载）
|
|
|
//输入： 1. src 输入图像结构体
|
|
|
// 2. degree 清晰度控制,有效范围【0,20】，值越大清晰度越好
|
|
|
|
|
|
//输出： 1. dst 输出图像结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoDefinitionCtrl(ImgStru* src,
|
|
|
ImgStru* dst,
|
|
|
int degree = 10);
|
|
|
|
|
|
|
|
|
// 5. 单帧图像目标定位 算法：王家星
|
|
|
|
|
|
//5.1 功能：计算图像中心点经纬度坐标、载荷等效俯仰角和方位角
|
|
|
//输入： 1. frame 输入复接数据
|
|
|
//
|
|
|
//输出： 1. Lon 图像中心点经度
|
|
|
// 2. Lat： 图像中心点纬度
|
|
|
// 3. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoCalImgCenterCoordinate(double &Lon,
|
|
|
double &Lat,
|
|
|
double &H,
|
|
|
const struQB_FJ *frame);
|
|
|
|
|
|
|
|
|
//5.2 功能：计算图像中任意点经纬度坐标
|
|
|
//输入： 1. frame 输入复接数据
|
|
|
// 2. pt_x： 列
|
|
|
// 3. pt_y： 行
|
|
|
//
|
|
|
//输出： 1. Lon 经度
|
|
|
// 2. Lat： 纬度
|
|
|
// 3. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoCalAnyPtCoordinate(double &Lon,
|
|
|
double &Lat,
|
|
|
double &H,
|
|
|
const struQB_FJ *frame,
|
|
|
int pt_x,
|
|
|
int pt_y);
|
|
|
|
|
|
|
|
|
// 6. 几何校正 算法：王家星
|
|
|
|
|
|
//6.1 功能：基于复接数据的图像系统级几何校正
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. bSrcFirst 默认时，优先对srcImg进行几何校正，否则对dstImg进行几何校正
|
|
|
|
|
|
//输出： 1. qbData 含系统级几何校正后图像的情报数据结构体
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoSysGeoCorrect(QBStru *qbData,
|
|
|
bool bSrcFirst = true);
|
|
|
|
|
|
|
|
|
//6.2 功能：基于复接数据的图像系统级几何校正（重载）
|
|
|
//输入： 1. srcImg 输入图像数据
|
|
|
// 2. qbData 输入复接数据
|
|
|
|
|
|
//输出： 1. dstImg 输出图像数据
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoSysGeoCorrect(ImgStru *dstImg,
|
|
|
const ImgStru *srcImg,
|
|
|
const struQB_FJ *qbData);
|
|
|
|
|
|
|
|
|
//6.3 功能：基于地面控制点的几何精校正
|
|
|
//输入： 1. srcImg 输入图像数据
|
|
|
// 2. GCP_XY 控制点（像素坐标）
|
|
|
// 3. GCP_LonLat 控制点（经纬度坐标）
|
|
|
// 4. type: 变换类型 -1：自适应 0：线性 1：二次多项式
|
|
|
|
|
|
//输出： 1. dstImg 输出图像数据
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoGCPGeoCorrectImg(ImgStru *dstImg,
|
|
|
const ImgStru *srcImg,
|
|
|
const std::vector<Point2d>& GCP_XY,
|
|
|
const std::vector<Point2d>& GCP_LonLat,
|
|
|
int type = -1);
|
|
|
|
|
|
|
|
|
//6.4 功能： OpenCV图像格式基于地面控制点的几何精校正
|
|
|
//输入： 1. srcImg 输入图像数据
|
|
|
// 2. GCP_XY 控制点（像素坐标）
|
|
|
// 3. GCP_LonLat 控制点（经纬度坐标）
|
|
|
// 4. type: 变换类型 -1：自适应 0：线性 1：二次多项式
|
|
|
|
|
|
//输出： 1. dstImg 输出图像数据
|
|
|
// 2. LBbox 经纬度边界
|
|
|
// 3. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoGCPGeoCorrectImg( cv::Mat& dstImg,
|
|
|
GeoBoundingBox& LBbox,
|
|
|
const cv::Mat& srcImg,
|
|
|
const std::vector<cv::Point2d>& GCP_XY,
|
|
|
const std::vector<cv::Point2d>& GCP_LonLat,
|
|
|
int type = -1);
|
|
|
|
|
|
|
|
|
//6.5 功能：基于控制点的几何校正
|
|
|
//输入： 1. srcImg 输入图像数据
|
|
|
// 2. GCP_xy 控制点（畸变像素坐标）
|
|
|
// 3. GCP_XY 控制点（参考图像像素坐标）
|
|
|
|
|
|
//输出： 1. dstImg 输出图像数据
|
|
|
// 2. 返回值：处理成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool DoCPBasedImgCorrect( cv::Mat &dstImg,
|
|
|
const cv::Mat &srcImg,
|
|
|
const std::vector<cv::Point2d>& GCP_xy,
|
|
|
const std::vector<cv::Point2d>& GCP_XY);
|
|
|
|
|
|
|
|
|
// 7. GeoTIFF文件生成保存 算法：王家星
|
|
|
|
|
|
//7.1 功能：保存情报数据至本地文件（GeoTiff格式）
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. savePath Tiff文件保存路径
|
|
|
|
|
|
//输出： 1. 返回值：保存成功返回true，失败返回false
|
|
|
//说明： 可能会先执行几何校正操作，再保存；
|
|
|
IMGPROCESS_API bool SaveToGeoTIFF(QBStru *qbData,
|
|
|
string savePath = "");
|
|
|
|
|
|
|
|
|
//7.2 功能：保存图像数据至本地文件（GeoTiff格式）
|
|
|
//输入： 1. Img 通用情报数据结构体
|
|
|
// 2. savePath Tiff文件保存路径
|
|
|
|
|
|
//输出： 1. 返回值：保存成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool SaveToGeoTIFF(const ImgStru *Img,
|
|
|
string savePath = "");
|
|
|
|
|
|
|
|
|
//7.3 功能：保存图像数据至本地文件（GeoTiff格式）
|
|
|
//输入： 1. Img OpenCV图像
|
|
|
// 2. LBbox 经纬度边界
|
|
|
// 3. savePath Tiff文件保存路径
|
|
|
|
|
|
|
|
|
//输出： 1. 返回值：保存成功返回true，失败返回false
|
|
|
IMGPROCESS_API bool SaveToGeoTIFF(cv::Mat& Img,
|
|
|
GeoBoundingBox& LBbox,
|
|
|
string savePath = "");
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 8. 距离、方位量算 算法：王家星
|
|
|
|
|
|
//8.1 功能：计算图像（几何校正前）中两点间的距离和方位（重载）
|
|
|
//输入： 1. frameData 通用复接数据结构体
|
|
|
// 2. pt1 起点（像素坐标）
|
|
|
// 3. pt2 终点（像素坐标）
|
|
|
|
|
|
//输出： 1. distance 距离
|
|
|
// 2. azAngle 方位，起点点到终点
|
|
|
// 3. 返回值：计算有效返回true，否则返回false
|
|
|
IMGPROCESS_API bool CalculateTwoPtsDistanceAzimuth( double &distance,
|
|
|
double &azAngle,
|
|
|
const struQB_FJ *frameData,
|
|
|
cv::Point pt1,
|
|
|
cv::Point pt2);
|
|
|
|
|
|
//8.2 功能：计算图像中两点间的距离和方位
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. pt1 起点（像素坐标）
|
|
|
// 3. pt2 终点（像素坐标）
|
|
|
// 4. flags：flags = 0：标识pt1、pt2为几何校正前图像中的点
|
|
|
// flags = 1：标识pt1、pt2为几何校正后图像中的点
|
|
|
|
|
|
//输出： 1. distance 距离
|
|
|
// 2. azAngle 方位，起点点到终点
|
|
|
// 3. 返回值：计算有效返回true，否则返回false
|
|
|
IMGPROCESS_API bool CalculateTwoPtsDistanceAzimuth( double &distance,
|
|
|
double &azAngle,
|
|
|
const QBStru *qbData,
|
|
|
cv::Point pt1,
|
|
|
cv::Point pt2,
|
|
|
int flags = 0);
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// 9. 面积量算 算法：王家星
|
|
|
|
|
|
//9.1 功能：计算图像（几何校正前）中多点组成的多边形面积
|
|
|
//输入： 1. frameData 复接数据
|
|
|
// 2. points 指向点坐标的指针
|
|
|
// 3. count： 点数目
|
|
|
|
|
|
//输出： 1. area ：多边形面积
|
|
|
// 2. 返回值：计算有效返回true，否则返回false
|
|
|
IMGPROCESS_API bool CalculatePloygonArea(double &area,
|
|
|
const struQB_FJ *frameData,
|
|
|
cv::Point *points,
|
|
|
int count);
|
|
|
|
|
|
//9.2 功能：计算图像中多边形面积
|
|
|
//输入： 1. qbData 通用情报数据结构体
|
|
|
// 2. point 多点指针（像素坐标）
|
|
|
// 3. count 点数目
|
|
|
// 4. flags：flags = 0：标识为几何校正前图像中的点
|
|
|
// flags = 1：标识为几何校正后图像中的点
|
|
|
|
|
|
//输出： 1. area 面积（平方米）
|
|
|
// 2. 返回值：计算有效返回true，否则返回false
|
|
|
//说明： 各点应为顺序或者逆序，不能乱序
|
|
|
IMGPROCESS_API bool CalculatePloygonArea(double &area,
|
|
|
const QBStru *qbData,
|
|
|
cv::Point* point,
|
|
|
int count,
|
|
|
int flags = 0);
|
|
|
|
|
|
//10 功能：基于经纬度读取地面高程
|
|
|
//输入：
|
|
|
// 1. fLonDeg: 经度（度）
|
|
|
// 2. fLatDeg: 纬度（度）
|
|
|
// 3. sDir: 高程数据文件存储路径， 默认设为："D:\\ElevationData"
|
|
|
//
|
|
|
//输出：
|
|
|
// 1. fElevator:高程（米）
|
|
|
//
|
|
|
//返回值：若函数运行成功，返回true，否则返回false
|
|
|
IMGPROCESS_API bool GetElevation( float& fElevator, double dLonDeg, double dLatDeg, string sDir, bool bNearst = false);
|
|
|
IMGPROCESS_API void ClearDem();
|
|
|
|
|
|
// 11 图像融合算法
|
|
|
|
|
|
// 11.1 功能：函数 DoFastFuseImgRGBplusIR() 快速融合两幅输入图像
|
|
|
// 输入：
|
|
|
// img1： 输入图像1：RGB：三通道
|
|
|
// img2： 输入图像2：IR：三通道或单通道
|
|
|
// 输出：
|
|
|
// dst: 融合图像
|
|
|
// 返回值：
|
|
|
// 运算有效返回true，否则返回false
|
|
|
IMGPROCESS_API bool DoFastFuseImgRGBplusIR(cv::Mat &dst, const cv::Mat RGBimg, const cv::Mat IRimg);
|
|
|
|
|
|
|
|
|
// 11.2 功能：函数 DoFuseImgRGBplusIR() 基于小波变换融合两幅输入图像
|
|
|
// 输入：
|
|
|
// img1： 输入图像1：RGB：三通道
|
|
|
// img2： 输入图像2：IR：三通道或单通道
|
|
|
// wname: 小波名称
|
|
|
// 支持小波：
|
|
|
// 1. haar
|
|
|
// 2. symN [1, 10]
|
|
|
// 3. dbN [1, 10]
|
|
|
// 4. coifN [1, 5]
|
|
|
// level：小波分解层数
|
|
|
// 输出：
|
|
|
// dst: 融合图像
|
|
|
// 返回值：
|
|
|
// 运算有效返回true，否则返回false
|
|
|
IMGPROCESS_API bool DoFuseImgRGBplusIR(cv::Mat &dst, const cv::Mat RGBimg, const cv::Mat IRimg, const string wname = "sym5", const unsigned int level = 10);
|
|
|
|
|
|
|
|
|
|
|
|
// 11.3 功能：函数 FuseImg() 基于小波变换融合两幅输入图像
|
|
|
// 输入：
|
|
|
// img1： 输入图像1
|
|
|
// img2： 输入图像2
|
|
|
// wname: 小波名称
|
|
|
// 支持小波：
|
|
|
// 1. haar
|
|
|
// 2. symN [1, 10]
|
|
|
// 3. dbN [1, 10]
|
|
|
// 4. coifN [1, 10]
|
|
|
// level：小波分解层数
|
|
|
// 输出：
|
|
|
// dst: 融合图像
|
|
|
// 返回值：
|
|
|
// 运算有效返回true，否则返回false
|
|
|
IMGPROCESS_API bool DoFuseImg(cv::Mat &dst, const cv::Mat img1, const cv::Mat img2, const string wname = "sym5", const unsigned int level = 10);
|
|
|
|
|
|
|
|
|
// 12 超分重建
|
|
|
// 功能：灰度图像（红外图像）超分辨率构建
|
|
|
// 输入：
|
|
|
// imgDeque： 输入图像序列，要求图像尺寸相同，类型为CV_8UC1
|
|
|
// 输出：
|
|
|
// dst: 输出超分图像
|
|
|
// 返回重建中有效的图像数目
|
|
|
IMGPROCESS_API int GetSuperResolutionImg(cv::Mat &dst, const std::deque<cv::Mat> &imgDeque, double multiple = 2);
|
|
|
|
|
|
|
|
|
#endif |