def draw(tile, streets, squared_images):
drawer = TileDrawer.from_tile(tile)
for street in streets:
drawer.draw_line(street.nodes[0], street.nodes[1])
for t in squared_images:
drawer.draw_box(t.getCentreNode(), 50)
return drawer
from src.detection.BoxWalker import BoxWalker
from src.base.Bbox import Bbox
from src.base.TileDrawer import TileDrawer
'''
This example visualizes the results of the boxwalker
'''
zurich_bellevue = Bbox.from_lbrt(8.814650, 47.222553, 8.825035, 47.228935) # Take the BBox you want
walker = BoxWalker(zurich_bellevue)
walker.load_convnet()
walker.load_tiles()
walker.load_streets()
walker.walk() # Walk through the streets. This could take some time...
crosswalkNodes = walker.plain_result # Takes all results found
#crosswalkNodes = walker.compared_with_osm_result # Takes only the results which are not already in OSM
# Draw and show the result
drawer = TileDrawer.from_tile(walker.tile)
for node in crosswalkNodes:
drawer.draw_point(node)
drawer.drawsection.save("boxsave.jpg")
drawer.drawsection.show()
def show(self):
drawer = TileDrawer.from_tile(self.tile)
for street in self.streets:
drawer.draw_line(street.nodes[0], street.nodes[1])
drawer.show()
