def cut_squared_images(streets, tile):
squared_list = []
for street in streets:
walker = StreetWalker.from_street_tile(street, tile, None)
squared = walker._get_squared_tiles(walker.street.nodes[0], walker.street.nodes[1])
squared_list += squared
return squared_list
def walk(self):
self.status_printer.start_walking()
ready_for_walk = (not self.tile is None) and (not self.streets is None) and (not self.convnet is None)
if(not ready_for_walk): raise Exception("Not ready for walk. Load tiles, streets and convnet first")
results = []
nb_images = 0
for i in range(len(self.streets)):
street = self.streets[i]
streetwalker = StreetWalker.from_street_tile(street, self.tile, self.convnet)
street_results = streetwalker.walk()
results += street_results
nb_images += streetwalker.nb_images
self.status_printer.set_state(i,len(results))
self.status_printer.end_walking(nb_images)
self.plain_result = self._merge_near_nodes(results)
self.compared_with_osm_result = self._compare_osm_with_detected_crosswalks(self.plain_result)
return self.compared_with_osm_result
def test_from_street_tile(self):
(tile, streets) = self.get_tile_streets(self.smallTestBbox())
walker = StreetWalker.from_street_tile(streets[0], tile, None)
self.assertIsNotNone(walker.tile)
self.assertIsNotNone(walker.street)
