|
@@ -50,24 +50,24 @@ public class SceneMarkShapeServiceImpl extends ServiceImpl<MarkShapeMapper, Scen
|
|
|
public void editTrainStatus(SceneMarkShapeParamVO param) {
|
|
|
SceneMarkShape byNumAndImagePath = findByNumAndImagePath(param.getNum(), param.getImagePath());
|
|
|
if (ObjectUtil.isNotNull(byNumAndImagePath)){
|
|
|
- byNumAndImagePath.setReDetect(0);
|
|
|
- byNumAndImagePath.setToDetect(1);
|
|
|
+ byNumAndImagePath.setReTrain(0);
|
|
|
+ byNumAndImagePath.setToTrain(1);
|
|
|
updateById(byNumAndImagePath);
|
|
|
}
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public List<SceneMarkShape> findByReTrainStatus(Integer reDetect){
|
|
|
+ public List<SceneMarkShape> findByReTrainStatus(Integer reTrain){
|
|
|
LambdaQueryWrapper<SceneMarkShape> wrapper = Wrappers.lambdaQuery();
|
|
|
- wrapper.eq(SceneMarkShape::getReDetect,reDetect);
|
|
|
+ wrapper.eq(SceneMarkShape::getReTrain,reTrain);
|
|
|
return list(wrapper);
|
|
|
}
|
|
|
|
|
|
@Override
|
|
|
- public List<SceneMarkShape> findByToTrainStatus(Integer toDetect) {
|
|
|
+ public List<SceneMarkShape> findByToTrainStatus(Integer toTrain) {
|
|
|
QueryWrapper queryWrapper = new QueryWrapper();
|
|
|
queryWrapper.select("DISTINCT num")
|
|
|
- .eq("to_detect",toDetect) ;
|
|
|
+ .eq("to_detect",toTrain) ;
|
|
|
return getBaseMapper().selectList(queryWrapper);
|
|
|
}
|
|
|
|
|
@@ -133,7 +133,7 @@ public class SceneMarkShapeServiceImpl extends ServiceImpl<MarkShapeMapper, Scen
|
|
|
}
|
|
|
log.info("转换labelimg标注处理的结果结束,{}",shapeJsons);
|
|
|
shape.setShapes(shapeJsons);
|
|
|
- shape.setReDetect(1);
|
|
|
+ shape.setReTrain(1);
|
|
|
shape.setUpdateTime(new Date());
|
|
|
return ResultData.ok(updateById(shape));
|
|
|
}else {
|