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)
