diff --git a/src/main/java/org/dromara/easyai/nerveEntity/ConvParameter.java b/src/main/java/org/dromara/easyai/nerveEntity/ConvParameter.java new file mode 100644 index 0000000..82db62e --- /dev/null +++ b/src/main/java/org/dromara/easyai/nerveEntity/ConvParameter.java @@ -0,0 +1,165 @@ +package org.dromara.easyai.nerveEntity; + +import org.dromara.easyai.matrixTools.Matrix; + +import java.util.ArrayList; +import java.util.HashMap; +import java.util.List; +import java.util.Map; + +/** + * @author lidapeng + * @time 2025/2/27 20:21 + * @des 卷积参数保存 + */ +public class ConvParameter { + private final List nerveMatrixList = new ArrayList<>();//下采样卷积权重矩阵 需取出模型 + private final List dymStudyRateList = new ArrayList<>();//动态学习率 + private final List convSizeList = new ArrayList<>(); + private final List im2colMatrixList = new ArrayList<>(); + private final List outMatrixList = new ArrayList<>(); + private List> oneConvPower;//1*1卷积核 需取出模型 + private List> oneDymStudyRateList;// + private List upOneDymStudyRateList; + private List upOneConvPower;//上采样降维卷积核 需要取出模型 + private List featureMatrixList;//所有通道特征矩阵集合 + private final List upDymStudyRateList = new ArrayList<>(); + private final List upNerveMatrixList = new ArrayList<>();//上采样卷积权重 需要取出模型 + private List upFeatureMatrixList;//上采样输入特征 + private List upOutMatrixList;//上卷积输出矩阵集合 + private final Map> featureMap = new HashMap<>(); + private int outX; + private int outY; + private int encoderX; + private int encoderY; + private float studyRateTh = 0;//记录线性模型动态偏移值学习率 + + public float getStudyRateTh() { + return studyRateTh; + } + + public void setStudyRateTh(float studyRateTh) { + this.studyRateTh = studyRateTh; + } + + public List getUpDymStudyRateList() { + return upDymStudyRateList; + } + + public List getUpOneDymStudyRateList() { + return upOneDymStudyRateList; + } + + public void setUpOneDymStudyRateList(List upOneDymStudyRateList) { + this.upOneDymStudyRateList = upOneDymStudyRateList; + } + + public List> getOneDymStudyRateList() { + return oneDymStudyRateList; + } + + public void setOneDymStudyRateList(List> oneDymStudyRateList) { + this.oneDymStudyRateList = oneDymStudyRateList; + } + + public List getDymStudyRateList() { + return dymStudyRateList; + } + + public List getUpFeatureMatrixList() { + return upFeatureMatrixList; + } + + public void setUpFeatureMatrixList(List upFeatureMatrixList) { + this.upFeatureMatrixList = upFeatureMatrixList; + } + + public List getUpOutMatrixList() { + return upOutMatrixList; + } + + public void setUpOutMatrixList(List upOutMatrixList) { + this.upOutMatrixList = upOutMatrixList; + } + + public List getUpOneConvPower() { + return upOneConvPower; + } + + public void setUpOneConvPower(List upOneConvPower) { + this.upOneConvPower = upOneConvPower; + } + + + public int getEncoderX() { + return encoderX; + } + + public void setEncoderX(int encoderX) { + this.encoderX = encoderX; + } + + public int getEncoderY() { + return encoderY; + } + + public void setEncoderY(int encoderY) { + this.encoderY = encoderY; + } + + public Map> getFeatureMap() { + return featureMap; + } + + public List getUpNerveMatrixList() { + return upNerveMatrixList; + } + + public int getOutX() { + return outX; + } + + public void setOutX(int outX) { + this.outX = outX; + } + + public int getOutY() { + return outY; + } + + public void setOutY(int outY) { + this.outY = outY; + } + + public List getNerveMatrixList() { + return nerveMatrixList; + } + + public List getConvSizeList() { + return convSizeList; + } + + public List getIm2colMatrixList() { + return im2colMatrixList; + } + + public List getOutMatrixList() { + return outMatrixList; + } + + public List> getOneConvPower() { + return oneConvPower; + } + + public void setOneConvPower(List> oneConvPower) { + this.oneConvPower = oneConvPower; + } + + public List getFeatureMatrixList() { + return featureMatrixList; + } + + public void setFeatureMatrixList(List featureMatrixList) { + this.featureMatrixList = featureMatrixList; + } +}