#ifndef __MATRIX_H__
#define __MATRIX_H__
#include <vector>
#include "Vriable.h"
#define DLength(x) (sizeof(x)/sizeof(double))
#define ILength(x) (sizeof(x)/sizeof(int))
namespace nn {
class MatCommaInitializer_;
class Matrix
{
public:
explicit Matrix();
/**
@brief 生成n*n*1的方阵,用0填充
@param n 矩阵行数
*/
Matrix(int n);
/**
@brief 生成row*col*1的方阵,用0填充
@param row 矩阵行数
@param col 矩阵列数
*/
Matrix(int row, int col);
/**
@brief 生成row*col*depth的方阵,用0填充
@param row 矩阵行数
@param col 矩阵列数
@param depth 矩阵通道数
*/
Matrix(int row, int col, int depth);
/**
@brief 生成size*1的方阵,用0填充
@param size_ 矩阵尺寸
*/
Matrix(Size size_);
/**
@brief 生成size的方阵,用0填充
@param size_ 矩阵尺寸
*/
Matrix(Size3 size_);
/**
@brief 拷贝函数
@param src 拷贝对象
*/
Matrix(const Matrix *src);
/**
拷贝函数
@param src 拷贝对象
*/
Matrix(const Matrix &src);
/**
@brief 将矩阵a和b合并(COL为按列合并|ROW为按行合并)
@param a 输入矩阵1
@param b 输入矩阵2
@param merge 合并方式
*/
Matrix(Matrix a, Matrix b, X_Y_Z merge);
/**
@brief 构造函数
深拷贝m
@param m 矩阵
*/
Matrix(MatCommaInitializer_ &m);
/**
@brief 生成n*n*1的方阵,元素为matrix
@param matrix 矩阵元素
@param n 矩阵大小
*/
Matrix(int *matrix, int n);
/**
@brief 生成n*n*1的方阵,元素为matrix
@param matrix 矩阵元素
@param n 矩阵大小
*/
Matrix(double *matrix, int n);
/**
@brief 生成row*col*1的方阵,元素为matrix
@param matrix 矩阵元素
@param row 矩阵行数
@param col 矩阵列数
*/
Matrix(int *matrix, int row, int col, int channel = 1);
/**
@brief 生成row*col*1的方阵,元素为matrix
@param matrix 矩阵元素
@param row 矩阵行数
@param col 矩阵列数
*/
Matrix(double *matrix, int row, int col, int channel = 1);
~Matrix();
/**
@brief 返回矩阵指针
*/
double* mat_()const;
/**
@brief 检查维度
*/
void DimCheck()const;
/**
@brief 返回矩阵尺寸(row,col,channel)
*/
Size3 size3()const;
/**
@brief 返回矩阵行数
*/
int rows()const;
/**
@brief 返回矩阵列数
*/
int cols()const;
/**
@brief 返回矩阵通道数
*/
int channels()const;
/**
@brief 返回矩阵大小(row*col*1)
*/
int size()const;
/**
@brief 返回矩阵大小Size(row,col)
*/
Size mSize()const;
/**
@brief 返回矩阵大小(row*col*1)
*/
int length()const;
/**
@brief 返回矩阵状态
0为方阵
-1为空矩阵
-2为非方阵
*/
int isEnable()const;
/**
@brief 返回矩阵是否为空
*/
bool empty()const;
/**
@brief 返回矩阵是否为方阵
*/
bool Square()const;
/**
@brief 按索引返回矩阵元素
@param index 索引
*/
double& at(int index)const;
/**
@brief 按索引返回矩阵元素
@param index_x 行索引
@param index_y 列索引
*/
double& at(int index_y, int index_x)const;
/**
@brief 将索引转换为对应矩阵列索引
@param index 索引
*/
int toX(int index)const;
/**
@brief 将索引转换为对应矩阵行索引
@param index 索引
*/
int toY(int index)const;
/**
@brief 矩阵第一个元素
*/
double frist()const;
/**
@brief 返回矩阵与value相等的第一个元素索引
@param value 元素
*/
int find(double value)const;
/**
@brief 返回矩阵元素最大值的索引
*/
int maxAt()const;
/**
@brief 返回矩阵元素最小值的索引
*/
int minAt()const;
/**
@brief 返回矩阵是否包含value
@param value 元素
*/
bool contains(double value)const;
/**
@brief 返回矩阵与value相等的第一个元素
@param value 元素
*/
double& findAt(double value)const;
/**
@brief 返回矩阵元素最大值
*/
double& findmax()const;
/**
@brief 返回矩阵元素最小值
*/
double& findmin()const;
/**
@brief 将矩阵按索引区域拷贝元素到src矩阵中
@param src 被拷贝矩阵
@param Row_Start 截取行初始索引值
@param Row_End 截取行结束索引值
@param Col_Start 截取列初始索引值
@param Col_End 截取列结束索引值
*/
void copy(Matrix &src, int Row_Start, int Row_End, int Col_Start, int Col_End)const;
/**
@brief 将矩阵拷贝到src
@param src 被拷贝矩阵
*/
void swap(Matrix &src)const;
/**
@brief 在矩阵最左边或最右边添加一列1
@param dire 选择添加方式
*/
void addones(direction dire = LEFT);
/**
@brief mChannel 将src覆盖到第channel通道
@param src 矩阵
@param channel 通道数
*/
void mChannel(const Matrix &src, int channel);
/**
@brief 设置矩阵维度
不允许改变矩阵长度
*/
void reshape(int row = 0, int col = 0, int channel = 0);
/**
@brief 设置矩阵大小
如果矩阵原大小不等于row*col*1则元素全部重置为0
@param row 矩阵行数
@param col 矩阵列数
*/
bool setSize(int row, int col);
/**
@brief 拷贝矩阵src
@param src 拷贝矩阵
*/
void setvalue(const Matrix &src);
/**
@brief 修改矩阵对应索引元素
@param number 元素
@param index 索引
*/
void setNum(double number, int index);
/**
@brief 修改矩阵对应索引元素
@param number 元素
@param index_y 行索引
@param index_x 列索引
*/
void setNum(double number, int index_y, int index_x);
/**
@brief 重置矩阵
@param mat 矩阵元素
@param row 矩阵行数
@param col 矩阵列数
*/
void setMat(double *mat, int row, int col);
/**
@brief 设置逆矩阵
*/
void setInv();
/**
@brief 设置矩阵的num次幂
@param num 次幂
*/
void setPow(int num);
/**
@brief 设置取反
*/
void setOpp();
/**
@brief 设置单位矩阵
*/
void setIden();
/**
@brief 设置伴随矩阵
*/
void setAdj();
/**
@brief 设置转置矩阵
*/
void setTran();
/**
@brief 命令行输出矩阵
*/
void show()const;
/**
@brief 返回取反矩阵
*/
const Matrix Opp()const;
/**
@brief 返回绝对值矩阵
*/
const Matrix Abs()const;
/**
@brief 返回按num次幂矩阵
@param num 次幂
*/
const Matrix Pow(int num)const;
/**
@brief 返回按元素取指数矩阵
*/
const Matrix Exp()const;
/**
@brief 返回按元素取对数矩阵
*/
const Matrix Log()const;
/**
@brief 返回按元素取开方矩阵
*/
const Matrix Sqrt()const;
/**
@brief 返回伴随矩阵
*/
const Matrix Adj()const;
/**
@brief 返回转置矩阵
*/
const Matrix t()const;
/**
@brief 返回逆矩阵
*/
const Matrix Inv()const;
/**
@brief 返回逆序矩阵
矩阵必须是向量
*/
const Matrix Reverse()const;
const Matrix EigenvectorsMax(double offset = 1e-8)const;
/**
@brief 返回行列式
*/
double Det();
/**
@brief 返回num范数
@param num 几范数
*/
double Norm(int num = 1)const;
/**
@brief 返回对应索引的余子式
@param x 列索引
@param y 行索引
*/
double Cof(int x, int y);
double EigenvalueMax(double offset = 1e-8)const;
/**
@brief 返回随机抽取的矩阵元素
*/
double RandSample();
/**
@brief 返回矩阵元素和
@param num 设置次幂
@param _abs 是否取绝对值
*/
double Sum(int num = 1, bool _abs = false)const;
/**
@brief 重载运算符+
对应元素相加
*/
const Matrix operator + (const double val)const;
/**
@brief 重载运算符+
对应元素相加
*/
const Matrix operator + (const Matrix &a)const;
/**
@brief 重载运算符+=
按元素相加
*/
void operator += (const double val);
/**
@brief 重载运算符+=
按元素相加
*/
void operator += (const Matrix &a);
/**
@brief 友元重载运算符+
按元素相加
*/
friend const Matrix operator + (const double value, const Matrix &mat);
/**
@brief 重载运算符-
按元素取相反数
*/
const Matrix operator - (void)const;
/**
@brief 重载运算符-
按元素相减
*/
const Matrix operator - (const double val)const;
/**
@brief 重载运算符-
对应元素相减
*/
const Matrix operator - (const Matrix &a)const;
/**
@brief 重载运算符-=
按元素相减
*/
void operator -= (const double val);
/**
@brief 重载运算符-=
对应元素相减
*/
void operator -= (const Matrix &a);
/**
@brief 友元重载运算符-
按元素相减
*/
friend const Matrix operator - (const double value, const Matrix &mat);
/**
@brief 重载运算符*
按元素相乘
*/
const Matrix operator * (const double val)const;
/**
@brief 重载运算符*
对应元素相乘
*/
const Matrix operator * (const Matrix &a)const;
/**
@brief 重载运算符*=
按元素相乘
*/
void operator *= (const double val);
/**
@brief 重载运算符*=
对应元素相乘
*/
void operator *= (const Matrix &a);
/**
@brief 友元重载运算符*
按元素相乘
*/
friend const Matrix operator * (const double value, const Matrix &mat);
/**
@brief 重载运算符/
按元素相除
*/
const Matrix operator / (const double val)const;
/**
@brief 重载运算符/
矩阵乘法
*/
const Matrix operator / (const Matrix &a)const;
/**
@brief 重载运算符/=
按元素相除
*/
void operator /= (const double val);
/**
@brief 重载运算符/=
对应元素相除
*/
void operator /= (const Matrix &a);
/**
@brief 友元重载运算符/
按元素相乘
*/
friend const Matrix operator / (const double value, const Matrix &mat);
/**
@brief 重载运算符=
深拷贝
*/
void operator = (const Matrix &temp);
/**
@brief 重载运算符==
判断矩阵是否相等
*/
bool operator == (const Matrix &a)const;
/**
@brief 重载运算符!=
判断矩阵是否不相等
*/
bool operator != (const Matrix &a)const;
/**
@brief 返回对应索引元素
@param index 索引
*/
double& operator () (const int index)const;
/**
@brief 返回对应索引元素
@param row 行索引
@param col 列索引
*/
double& operator () (const int row, const int col)const;
/**
@brief 返回对应索引元素
@param row 行索引
@param col 列索引
@param depth 通道索引
*/
double& operator () (const int row, const int col, const int depth)const;
/**
@brief 返回矩阵对应索引的列或行
@param index 索引
@param rc 索引方式
*/
const Matrix operator () (const int index, X_Y_Z rc)const;
/**
@brief 返回矩阵对应通道索引
@param channel 通道索引
*/
const Matrix operator [] (const int channel)const;
friend std::ostream & operator << (std::ostream &out, const Matrix &ma);
private:
int row;
int col;
int depth;
bool square;
double *matrix;
void init();
void checkSquare();
void checkindex(int index)const;
void checkindex(int index_x, int index_y)const;
};
typedef Matrix Mat;
/**
@brief Mat_ 工具类
继承Mat类，用于实现
Mat mat = (Mat_(3, 3) <<
-1, -1, -1,
-1, 9, -1,
-1, -1, -1);
*/
class Mat_ : public Mat
{
public:
explicit Mat_() {}
/**
@brief 生成row*col*channel的方阵,用0填充
@param row 矩阵行数
@param col 矩阵列数
@param depth 矩阵通道数
*/
Mat_(int row, int col = 1, int channel = 1) : Mat(row, col, channel) {}
/**
@brief 生成size_[0]*size_[1]*size_[2]的方阵,用0填充
会释放Vec<int> size_的内存
@param size_ 矩阵尺寸
*/
Mat_(const Size3 &size_) : Mat(size_) {}
};
/**
@brief MatCommaInitializer_ 工具类
作为迭代器，用于实现
Mat mat = (Mat_(3, 3) <<
-1, -1, -1,
-1, 9, -1,
-1, -1, -1);
*/
class MatCommaInitializer_
{
public:
explicit MatCommaInitializer_() {}
MatCommaInitializer_(const Mat_& m) {
head = m.mat_();
it = head;
row = m.rows();
col = m.cols();
channel = m.channels();
}
template<typename Tp_>
MatCommaInitializer_ operator , (Tp_ v);
int rows()const { return row; }
int cols()const { return col; }
int channels()const { return channel; }
double * matrix()const { return head; }
private:
int row;
int col;
int channel;
double *it;
double *head;
};
template<typename Tp_>
inline MatCommaInitializer_ MatCommaInitializer_::operator , (Tp_ v)
{
if (this->it == this->head + row*col*channel) {
fprintf(stderr, errinfo[ERR_INFO_MEMOUT]);
throw MatCommaInitializer_();
}
*this->it = double(v);
++this->it;
return *this;
}
template<typename Tp_>
static MatCommaInitializer_ operator << (const Mat_& m, Tp_ val)
{
MatCommaInitializer_ commaInitializer(m);
return (commaInitializer, val);
}
void check(int row, int col, int depth = 1);
/**
@brief 设置随机数种子
*/
void Srandom();
/**
@brief 返回矩阵的最大值
@param src 矩阵
@param isAbs 是否取绝对值
*/
double Max(const Matrix &src, bool isAbs = false);
/**
@brief 返回矩阵的最小值
@param src 矩阵
@param isAbs 是否取绝对值
*/
double Min(const Matrix &src, bool isAbs = false);
/**
@brief 返回矩阵的行列式
@param src 矩阵
*/
double det(const Matrix &src);
/**
@brief 返回矩阵的迹
@param src 矩阵
*/
double trace(const Matrix &src);
/**
@brief 返回矩阵对应索引的余子式
@param src 矩阵
@param x 列索引
@param y 行索引
*/
double cof(const Matrix &src, int x, int y);
/**
@brief 返回随机数
@param min 最小值
@param max 最大值
@param isdouble 是否随机非整数
*/
double getRandData(int min, int max, bool isdouble = false);
/**
@brief 返回矩阵之和
@param src 矩阵
*/
double mSum(const Matrix &src);
/**
@brief 返回矩阵范数
@param src 矩阵
@param num 几范数
*/
double mNorm(const Matrix& src, int num = 1);
/**
@brief 返回矩阵的距离
@param a 矩阵
@param b 矩阵
@param num 几范数
*/
double mDistance(const Matrix& a, const Matrix& b, int num = 2);
/**
@brief 返回随机的矩阵元素
@param src 矩阵
*/
double mRandSample(const Matrix &src);
/**
@brief 返回将向量转换成1*point.size()*1的矩阵
@param point 向量
*/
const Matrix VectoMat(std::vector<double> &point);
/**
@brief 返回将向量转换成point.size()*point[0].size()*1的矩阵
@param points 向量
*/
const Matrix VectoMat(std::vector<std::vector<double>> &points);
/**
@brief 返回将矩阵转换成一维向量
@param src 矩阵
*/
std::vector<double> MattoVec(const Matrix &src);
/**
@brief 返回将矩阵转换成row维向量
@param src 矩阵
*/
std::vector<std::vector<double>> MattoVecs(const Matrix &src);
/**
@brief 返回生成的n*n*1单位矩阵
@param n 矩阵大小
*/
const Matrix eye(int n);
/**
@brief 返回矩阵的第channel的通道
@param src 矩阵
@param channel 通道索引
*/
const Matrix mSplit(const Matrix &src, int channel);
/**
@brief 返回按矩阵的通道数复制
@param src 输入矩阵
@param dst 输入矩阵通道数的矩阵数组
*/
void mSplit(const Matrix &src, Matrix *dst);
/**
@brief 返回按通道合并的矩阵
@param src 矩阵数组
@param channels 通道数
*/
const Matrix mMerge(const Matrix *src, int channels);
/**
@brief 返回按索引区域切割的矩阵
@param src 矩阵
@param Row_Start 截取行初始索引值
@param Row_End 截取行结束索引值
@param Col_Start 截取列初始索引值
@param Col_End 截取列结束索引值
*/
const Matrix Block(const Matrix &src, int Row_Start, int Row_End, int Col_Start, int Col_End);
/**
@brief 返回随机生成元素n*n*1矩阵
@param n 矩阵大小
@param low 下界
@param top 上界
@param isdouble 是否生成浮点数
*/
const Matrix mRand(int low, int top, int n, bool isdouble = false);
/**
@brief 返回随机生成元素size*1矩阵
@param low 下界
@param top 上界
@param size 矩阵大小
@param isdouble 是否生成浮点数
*/
const Matrix mRand(int low, int top, Size size, bool isdouble = false);
/**
@brief 返回随机生成元素size矩阵
@param low 下界
@param top 上界
@param size 矩阵大小
@param isdouble 是否生成浮点数
*/
const Matrix mRand(int low, int top, Size3 size, bool isdouble = false);
/**
@brief 返回随机生成元素row*col*channel矩阵
@param row 矩阵行数
@param col 矩阵列数
@param low 下界
@param top 上界
@param isdouble 是否生成浮点数
*/
const Matrix mRand(int low, int top, int row, int col, int channel = 1,bool isdouble = false);
/**
@brief 返回元素为0的row*col*1矩阵
@param row 矩阵行数
@param col 矩阵列数
*/
const Matrix zeros(int row, int col);
/**
@brief 返回元素为0的row*col*channel矩阵
@param row 矩阵行数
@param col 矩阵列数
@param channel 矩阵通道数
*/
const Matrix zeros(int row, int col, int channel);
/**
@brief 返回元素为0的size矩阵
@param size 矩阵大小
*/
const Matrix zeros(Size size);
/**
@brief 返回元素为0的size矩阵
@param size 矩阵大小
*/
const Matrix zeros(Size3 size);
/**
@brief 返回元素为v的row*col*channel矩阵
@param v 填充元素
@param row 矩阵行数
@param col 矩阵列数
@param channel 矩阵通道数
*/
const Matrix value(double v, int row, int col, int channel = 1);
/**
@brief 返回元素为1的row*col*1矩阵
@param row 矩阵行数
@param col 矩阵列数
@param v 填充元素
*/
const Matrix ones(int row, int col);
/**
@brief 返回元素为0的row*col*channel矩阵
@param row 矩阵行数
@param col 矩阵列数
@param channel 矩阵通道数
*/
const Matrix ones(int row, int col, int channel);
/**
@brief 返回元素为0的size矩阵
@param size 矩阵大小
*/
const Matrix ones(Size size);
/**
@brief 返回逆序矩阵，矩阵需为一维向量
@param src 矩阵
*/
const Matrix reverse(const Matrix &src);
/**
@brief 返回生成row*col*channel的矩阵，随机抽取矩阵src元素填充
@param src 矩阵
@param row 矩阵行数
@param col 矩阵列数
*/
const Matrix mRandSample(const Matrix &src, int row, int col, int channel = 1);
/**
@brief 返回随机抽取num次矩阵src的行或列组成的矩阵
@param src 矩阵
@param rc 抽取方式
@param num 抽取次数
*/
const Matrix mRandSample(const Matrix &src, X_Y_Z rc, int num = 1);
/**
@brief 返回从low到top按gap间隔分割的矩阵
@param low 下界
@param top 上界
@param gap 间隔
*/
const Matrix range(int low, int top, double gap = 1);
/**
@brief 返回从low到top按gap间隔分割的矩阵
@param low 下界
@param top 上界
@param gap 间隔
*/
const Matrix range(double low, double top, double gap = 1);
/**
@brief 返回从low到top等分成的1*len的矩阵
@param low 下界
@param top 上界
@param len 等分个数
*/
const Matrix linspace(int low, int top, int len);
/**
@brief 返回从low到top等分成的1*len的矩阵
@param low 下界
@param top 上界
@param len 等分个数
*/
const Matrix linspace(double low, double top, int len);
/**
@brief 返回矩阵的伴随矩阵
@param src 矩阵
*/
const Matrix adj(const Matrix &src);
/**
@brief 返回矩阵的逆矩阵
@param src 矩阵
*/
const Matrix inv(const Matrix &src);
/**
@brief 返回矩阵的伪逆矩阵
@param src 矩阵
@param dire 伪逆矩阵的计算方式
*/
const Matrix pinv(const Matrix &src, direction dire = LEFT);
/**
@brief 返回矩阵的转置矩阵
@param src 矩阵
*/
const Matrix tran(const Matrix &src);
/**
@brief 返回矩阵的绝对值矩阵
@param src 矩阵
*/
const Matrix mAbs(const Matrix &src);
/**
@brief 返回angle度2*2的旋转矩阵
@param angle 角度
*/
const Matrix Rotate(double angle);
/**
@brief 返回矩阵num次幂
@param src 矩阵
@param num 次幂
*/
const Matrix POW(const Matrix &src, int num);
/**
@brief 返回矩阵取反
@param src 矩阵
*/
const Matrix mOpp(const Matrix &src);
/**
@brief 返回矩阵按行或列之和
@param src 矩阵
@param rc 求和的方向
*/
const Matrix mSum(const Matrix &src, X_Y_Z rc);
/**
@brief 返回矩阵按元素取指数
@param src 矩阵
*/
const Matrix mExp(const Matrix &src);
/**
@brief 返回矩阵按元素取对数
@param src 矩阵
*/
const Matrix mLog(const Matrix &src);
/**
@brief 返回矩阵按元素取开方
@param src 矩阵
*/
const Matrix mSqrt(const Matrix &src);
/**
@brief 返回矩阵按元素取num次幂
@param src 矩阵
@param num 次幂
*/
const Matrix mPow(const Matrix &src, double num);
/**
@brief 返回矩阵val/src按元素除
@param src 矩阵
@param val 除数
*/
const Matrix Divi(const Matrix &src, double val, direction dire = RIGHT);
/**
@brief 返回矩阵除法
@param a 被除矩阵
@param b 除矩阵
@param dire 除法方式
*/
const Matrix Divi(const Matrix &a, const Matrix &b, direction dire = RIGHT);
/**
@brief 返回矩阵按元素对乘
@param a 矩阵
@param b 矩阵
*/
const Matrix Mult(const Matrix &a, const Matrix &b);
/**
@brief 返回矩阵按元素取a和b之间的最大值
@param a 比较值
@param b 比较矩阵
*/
const Matrix mMax(double a, const Matrix &b);
/**
@brief 返回矩阵按元素取a和b之间的最大值
@param a 比较矩阵
@param b 比较矩阵
*/
const Matrix mMax(const Matrix &a, const Matrix &b);
/**
@brief 返回矩阵按元素取a和b之间的最小值
@param a 比较值
@param b 比较矩阵
*/
const Matrix mMin(double a, const Matrix &b);
/**
@brief 返回矩阵按元素取a和b之间的最小值
@param a 比较矩阵
@param b 比较矩阵
*/
const Matrix mMin(const Matrix &a, const Matrix &b);
/**
@brief mCalSize 计算卷积所需扩张的边界
返回矩阵大小
@param src 被卷积矩阵
@param kern 卷积核
@param anchor 像素对应卷积核坐标
anchor默认为Point(-1,-1), 像素对应卷积核中心
@param strides 滑动步长
@param top 向上扩充几行
@param bottom 向下扩充几行
@param left 向左扩充几列
@param right 向右扩充几列
*/
Size3 mCalSize(const Matrix &src, const Matrix &kern, Point & anchor, Size strides, int &top, int &bottom, int &left, int &right);
/**
@brief 返回按boundary分界填充的矩阵
返回矩阵大小等于输入矩阵大小
@param src 输入矩阵
@param boundary 分界值
@param lower 小于boundary用lower填充
@param upper 大于boundary用upper填充
@param boundary2upper 当矩阵元素等于boundary时
为1归upper, 为-1归lower, 为0不处理
*/
const Matrix mThreshold(const Matrix &src, double boundary, double lower, double upper, int boundary2upper = 1);
/**
@brief 返回边界扩充的矩阵
@param src 输入矩阵
@param top 向上扩充几行
@param bottom 向下扩充几行
@param left 向左扩充几列
@param right 向右扩充几列
@param borderType 边界像素外推的插值方法
@param value 常量插值的数值
**/
const Matrix copyMakeBorder(const Matrix &src, int top, int bottom, int left, int right, BorderTypes borderType = BORDER_CONSTANT, const int value = 0);
/**
@brief 返回矩阵2维卷积结果
返回矩阵大小为(input.row/strides_x, input.col/strides_y, 1)
@param input 输入矩阵
@param kern 卷积核
@param anchor 矩阵元素对应卷积核的位置
以卷积核的左上角为(0,0)点, 默认(-1,-1)为中心
@param strides 滑动步长
Size.hei为x轴,Size.wid为y轴
@param is_copy_border 是否要扩展边界
*/
const Matrix Filter2D(const Mat & input, const Mat & kern, Point anchor = Point(-1, -1), const Size & strides = Size(1, 1), bool is_copy_border = true);
/**
@brief 命令行按矩阵输出
@param row 行
@param col 列
*/
template<typename T>
void showMatrix(const T *, int row, int col);
void pause();
}
#endif //__MATRIX_H__
main.cpp
#include <iostream>
#include "Mat.h"
using namespace std;
using namespace nn;
int main()
{
double data[] = {
-1, -1, -1,
0, 0, 0,
1, 1, 1
};
Mat kern_y(data, 3, 3);
cout << "kern_y = \n"<< kern_y << endl;
Mat kern_x = (Mat_(3, 3) <<
-1, 0, 1,
-1, 0, 1,
-1, 0, 1);
cout << "kern_x = \n" << kern_x << endl;
Mat x = zeros(3, 3);
cout << "zeros(3, 3) = \n" << x << endl;
Mat xx = range(-3, 3);
cout << "range(-3, 3) = \n" << xx << endl;
Mat x3 = linspace(-3, 3, 10);
cout << "linspace(-3, 3, 10) = \n" << x3 << endl;
Mat x4(kern_x, kern_y, ROW);
cout << "Mat x4(kern_x, kern_y, ROW) = \n" << x4 << endl;
Mat x5(x4, x4, COL);
cout << "Mat(x4, x4, COL) = \n" << endl;
x5.show(); cout << endl;
Mat x1 = mRand(1, 10, Size(3, 3), true);
cout << "x1 = mRand(1, 10, Size(3, 3), true) = \n" << x1 << endl;
Mat x2 = mRand(-10, 10, Size(3, 3));
cout << "x2 = mRand(-10, 10, Size(3, 3)) = \n" << x2 << endl;
cout << "x1.Det() = " << x1.Det() << endl;
cout << "x1.t() = \n" << x1.t() << endl;
cout << "x1.Adj() = \n" << x1.Adj() << endl;
cout << "x1.Inv() = \n" << x1.Inv() << endl;
cout << "x1.reshape(1, -1) = " << endl;
Mat x_r = x1;
x_r.reshape(1, -1);
cout << x_r << endl;
cout << "x1 + x2 = \n" << x1 + x2 << endl;
cout << "x1 - x2 = \n" << x1 - x2 << endl;
cout << "x1 * x2 = \n" << x1 * x2 << endl;
cout << "x1 / x2 = \n" << x1 / x2 << endl;
cout << "exp(x1) = \n" << mExp(x1) << endl;
cout << "log(x1) = \n" << mLog(x1) << endl;
cout << "mul(x1, x2) = \n" << Mult(x1, x2) << endl;
cout << "sum(x1) = " << mSum(x1) << endl;
cout << "norm(x1) = " << mNorm(x1) << endl;
cout << "norm(x1, 2) = " << mNorm(x1, 2) << endl;
cout << "abs(x1) = \n" << mAbs(x1) << endl;
cout << "mSqrt(x1) = \n" << mSqrt(x1) << endl;
cout << "mPow(x1, 1/2) = \n" << mPow(x1, 1/2.0) << endl;
cout << "mPow(x1, 2) = \n" << mPow(x1, 2) << endl;
cout << "adj(x1) = \n" << adj(x1) << endl;
pause();
return 0;
}
9024
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
