def process_image_for_vehicles(self, image):
vehicle_heatmap = find_cars(image)
imageData = ImageData(image, None, None, None, None)
imageData.set_vehicle_heatmap(vehicle_heatmap)
self.__save_image_data(imageData)
self.draw_vehicle_boxes(image)
return image
def process_image(self, image):
undist = self.__undistort(image)
vehicle_heatmap = find_cars(image)
# binary = self.__transform_to_binary(image)
# binary_warped = self.__warp_image(binary)
imageData = ImageData(image, None, None, None, None)
# imageData = None
# if (self.__last_image_data):
# imageData = poly_search(binary_warped, self.__last_image_data.left_fit(), self.__last_image_data.right_fit())
# if (not imageData):
# imageData = full_search(binary_warped)
imageData.set_vehicle_heatmap(vehicle_heatmap)
self.__save_image_data(imageData)
# result = self.__draw_lane_box(undist, binary_warped)
# self.draw_vehicle_boxes(result)
# self.__write_info(result)
self.draw_vehicle_boxes(image)
return image