diff --git a/src/main/java/org/dromara/easyai/resnet/ResConvPower.java b/src/main/java/org/dromara/easyai/resnet/ResConvPower.java new file mode 100644 index 0000000..8afb73b --- /dev/null +++ b/src/main/java/org/dromara/easyai/resnet/ResConvPower.java @@ -0,0 +1,64 @@ +package org.dromara.easyai.resnet; + + +import org.dromara.easyai.resnet.entity.DoubleResConvModel; + +import java.util.List; + +/** + * @author lidapeng + * @time 2025/4/11 13:48 + * @des 残差块卷积权重 + */ +public class ResConvPower { + private ConvLay firstConvPower;//第一层卷积权重 + private ConvLay secondConvPower;//第二层卷积权重 + private List> oneConvPower;//1v1卷积层 需要作为模型取出 + private List> dymStudyRateList;//梯度平方和 + + public List> getDymStudyRateList() { + return dymStudyRateList; + } + + public void setDymStudyRateList(List> dymStudyRateList) { + this.dymStudyRateList = dymStudyRateList; + } + + public DoubleResConvModel getModel() { + DoubleResConvModel model = new DoubleResConvModel(); + model.setFirstResConvModel(firstConvPower.getModel()); + model.setSecondResConvModel(secondConvPower.getModel()); + model.setOneConvPowerModel(oneConvPower); + return model; + } + + public void insertModel(DoubleResConvModel model) { + firstConvPower.insertModel(model.getFirstResConvModel()); + secondConvPower.insertModel(model.getSecondResConvModel()); + oneConvPower = model.getOneConvPowerModel(); + } + + public ConvLay getFirstConvPower() { + return firstConvPower; + } + + public void setFirstConvPower(ConvLay firstConvPower) { + this.firstConvPower = firstConvPower; + } + + public ConvLay getSecondConvPower() { + return secondConvPower; + } + + public void setSecondConvPower(ConvLay secondConvPower) { + this.secondConvPower = secondConvPower; + } + + public List> getOneConvPower() { + return oneConvPower; + } + + public void setOneConvPower(List> oneConvPower) { + this.oneConvPower = oneConvPower; + } +}