(no title)
daavoo | 11 months ago
Hi there! There is no SAM2 finetuning involved in the project. The segmentation data from OSM doesn't have enough quality to properly train a segmentation model from it.
What I am using here is a YOLO model for bounding box prediction. The bounding boxes from OSM are good enough for this. I then pass the each individual bounding box as a "prompt" to SAM2 for a segmentation of what is inside.
I also tried to pass the centroid of the box as a "prompt" for SAM but it gave worse results.
No comments yet.