def test_yield_sign(self):
image_name = "scene_all_signs.png"
test_image = cv2.imread("input_images/test_images/" + image_name)
coords = ps2.yield_sign_detection(test_image)
cv2.circle(test_image, coords, 5, (0, 255, 0), 2)
cv2.imshow('All_signs', test_image)
cv2.waitKey(0)
check_result(image_name, coords, (507, 333), 5)
def test_yield_sign(self):
image_name = "yield_173_358_background.png"
test_image = cv2.imread("input_images/test_images/" + image_name)
coords = ps2.yield_sign_detection(test_image)
check_result(image_name, coords, (358, 173), 5)