def part_5a(): input_images = ['img-5-a-1', 'img-5-a-2', 'img-5-a-3'] output_labels = ['ps2-5-a-1', 'ps2-5-a-2', 'ps2-5-a-3'] for img_in, label in zip(input_images, output_labels): scene = cv2.imread("input_images/{}.png".format(img_in)) coords = ps2.traffic_sign_detection_challenge(scene) img_out = mark_traffic_signs(scene, coords) cv2.imwrite("output/{}.png".format(label), img_out)
def part_5a(): input_images = ['img-5-a-1', 'img-5-a-2', 'img-5-a-3'] output_labels = ['ps2-5-a-1', 'ps2-5-a-2', 'ps2-5-a-3'] for img_in, label in zip(input_images, output_labels): scene = cv2.imread("input_images/{}.jpg".format(img_in)) x, y, _ = scene.shape y = int(y/(float(x)/500)) scene = cv2.resize(scene, (y, 500)) coords = ps2.traffic_sign_detection_challenge(scene) img_out = mark_traffic_signs(scene, coords) cv2.imwrite("output/{}.png".format(label), img_out)