增加边框模型参数的获取和注入

pull/1/head
lidapeng 5 years ago
parent 6b6fd14ee1
commit 932c4fa02d

@ -5,7 +5,6 @@ import org.wlld.MatrixTools.MatrixOperation;
import org.wlld.config.StudyPattern;
import org.wlld.function.ReLu;
import org.wlld.function.Sigmod;
import org.wlld.i.OutBack;
import org.wlld.imageRecognition.border.BorderBody;
import org.wlld.imageRecognition.border.Frame;
import org.wlld.nerveCenter.NerveManager;
@ -188,6 +187,10 @@ public class TempleConfig {
modelParameter.setDymNerveStudies(modelParameter1.getDymNerveStudies());
modelParameter.setDymOutNerveStudy(modelParameter1.getDymOutNerveStudy());
}
if (isHavePosition) {//存在边框学习模型参数
modelParameter.setBorderBodyMap(borderBodyMap);
modelParameter.setFrame(frame);
}
return modelParameter;
}
@ -227,6 +230,13 @@ public class TempleConfig {
if (studyPattern == StudyPattern.Accuracy_Pattern) {
convolutionNerveManager.insertModelParameter(modelParameter);
}
Frame frame = modelParameter.getFrame();
Map<Integer, BorderBody> borderBodyMap = modelParameter.getBorderBodyMap();
if (frame != null && borderBodyMap != null && borderBodyMap.size() > 0) {
isHavePosition = true;
this.frame = frame;
this.borderBodyMap = borderBodyMap;
}
}
public void setCutThreshold(double cutThreshold) {

@ -1,9 +1,13 @@
package org.wlld.nerveEntity;
import org.wlld.MatrixTools.Matrix;
import org.wlld.imageRecognition.border.BorderBody;
import org.wlld.imageRecognition.border.Frame;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
/**
* @author lidapeng
@ -16,6 +20,24 @@ public class ModelParameter {
private List<NerveStudy> outNevers = new ArrayList<>();//输出神经元
private List<DymNerveStudy> dymNerveStudies = new ArrayList<>();//动态神经元隐层
private DymNerveStudy dymOutNerveStudy = new DymNerveStudy();//动态神经元输出层
private Map<Integer, BorderBody> borderBodyMap = new HashMap<>();//border特征集合
private Frame frame;//先验边框
public Map<Integer, BorderBody> getBorderBodyMap() {
return borderBodyMap;
}
public void setBorderBodyMap(Map<Integer, BorderBody> borderBodyMap) {
this.borderBodyMap = borderBodyMap;
}
public Frame getFrame() {
return frame;
}
public void setFrame(Frame frame) {
this.frame = frame;
}
public List<DymNerveStudy> getDymNerveStudies() {
return dymNerveStudies;

@ -8,7 +8,6 @@ import org.wlld.imageRecognition.Operation;
import org.wlld.imageRecognition.Picture;
import org.wlld.imageRecognition.TempleConfig;
import org.wlld.nerveEntity.ModelParameter;
import org.wlld.test.Ma;
import java.util.HashMap;
import java.util.Map;

@ -1,4 +1,4 @@
package org.wlld.test;
package org.wlld;
import org.wlld.MatrixTools.Matrix;
import org.wlld.i.OutBack;
Loading…
Cancel
Save