use of boofcv.alg.feature.detect.line.gridline.GridLineModelFitter in project BoofCV by lessthanoptimal.
the class VisualizeLineRansac method process.
public void process(@Nullable BufferedImage image) {
Objects.requireNonNull(image);
// int regionSize = 40;
I input = GeneralizedImageOps.createSingleBand(imageType, image.getWidth(), image.getHeight());
D derivX = GeneralizedImageOps.createSingleBand(derivType, image.getWidth(), image.getHeight());
D derivY = GeneralizedImageOps.createSingleBand(derivType, image.getWidth(), image.getHeight());
GrayF32 edgeIntensity = new GrayF32(input.width, input.height);
// GrayF32 suppressed = new GrayF32(input.width,input.height);
// GrayF32 orientation = new GrayF32(input.width,input.height);
// GrayS8 direction = new GrayS8(input.width,input.height);
GrayU8 detected = new GrayU8(input.width, input.height);
ModelManager<LinePolar2D_F32> manager = new ModelManagerLinePolar2D_F32();
ModelMatcherPost<LinePolar2D_F32, Edgel> matcher = new Ransac<>(123123, 25, 1, manager, Edgel.class);
matcher.setModel(() -> new GridLineModelFitter((float) (Math.PI * 0.75)), () -> new GridLineModelDistance((float) (Math.PI * 0.75)));
ImageGradient<I, D> gradient = FactoryDerivative.sobel(imageType, derivType);
System.out.println("Image width " + input.width + " height " + input.height);
ConvertBufferedImage.convertFromSingle(image, input, imageType);
gradient.process(input, derivX, derivY);
GGradientToEdgeFeatures.intensityAbs(derivX, derivY, edgeIntensity);
// non-max suppression on the lines
// GGradientToEdgeFeatures.direction(derivX,derivY,orientation);
// GradientToEdgeFeatures.discretizeDirection4(orientation,direction);
// GradientToEdgeFeatures.nonMaxSuppression4(edgeIntensity,direction,suppressed);
GThresholdImageOps.threshold(edgeIntensity, detected, 30, false);
GridRansacLineDetector<GrayF32> alg = new ImplGridRansacLineDetector_F32(40, 10, matcher);
alg.process((GrayF32) derivX, (GrayF32) derivY, detected);
MatrixOfList<LineSegment2D_F32> gridLine = alg.getFoundLines();
// ConnectLinesGrid connect = new ConnectLinesGrid(Math.PI*0.01,1,8);
// connect.process(gridLine);
// LineImageOps.pruneClutteredGrids(gridLine,3);
List<LineSegment2D_F32> found = gridLine.createSingleList();
System.out.println("size = " + found.size());
LineImageOps.mergeSimilar(found, (float) (Math.PI * 0.03), 5f);
// LineImageOps.pruneSmall(found,40);
System.out.println("after size = " + found.size());
ImageLinePanel gui = new ImageLinePanel();
gui.setImage(image);
gui.setLineSegments(found);
gui.setPreferredSize(new Dimension(image.getWidth(), image.getHeight()));
BufferedImage renderedBinary = VisualizeBinaryData.renderBinary(detected, false, null);
ShowImages.showWindow(renderedBinary, "Detected Edges");
ShowImages.showWindow(gui, "Detected Lines");
}
use of boofcv.alg.feature.detect.line.gridline.GridLineModelFitter in project BoofCV by lessthanoptimal.
the class CommonGridRansacLineDetectorChecks method checkObvious.
/**
* Give it a single straight line and see if it can detect it. Allow the region size to be changed to check
* for issues related to that
*/
protected void checkObvious(int regionSize) {
// System.out.println("regionSize = "+regionSize);
int where = 25;
GrayU8 edgeImage = new GrayU8(width, height);
D derivX = GeneralizedImageOps.createSingleBand(derivType, width, height);
D derivY = GeneralizedImageOps.createSingleBand(derivType, width, height);
for (int i = 0; i < height; i++) {
edgeImage.set(where, i, 1);
GeneralizedImageOps.set(derivX, where, i, 20);
}
ModelManagerLinePolar2D_F32 manager = new ModelManagerLinePolar2D_F32();
ModelMatcherPost<LinePolar2D_F32, Edgel> matcher = new Ransac<>(123123, 25, 1, manager, Edgel.class);
matcher.setModel(() -> new GridLineModelFitter(0.9f), () -> new GridLineModelDistance(0.9f));
GridRansacLineDetector<D> alg = createDetector(regionSize, 5, matcher);
alg.process(derivX, derivY, edgeImage);
MatrixOfList<LineSegment2D_F32> lines = alg.getFoundLines();
assertEquals(width / regionSize, lines.getWidth());
assertEquals(height / regionSize, lines.getHeight());
int gridCol = where / regionSize;
for (int i = 0; i < lines.height; i++) {
List<LineSegment2D_F32> l = lines.get(gridCol, i);
assertEquals(l.size(), 1);
LineSegment2D_F32 a = l.get(0);
assertTrue(Math.abs(a.slopeY()) > 1);
assertTrue(Math.abs(a.slopeX()) < 0.01);
}
}
Aggregations