def test_warning_sign(self):
image_name = "scene_all_signs.png"
test_image = cv2.imread("input_images/test_images/" + image_name)
coords = ps2.warning_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, (799, 349), 5)
def test_warning_sign(self):
image_name = "warning_250_300_background.png"
test_image = cv2.imread("input_images/test_images/" + image_name)
coords = ps2.warning_sign_detection(test_image)
check_result(image_name, coords, (300, 250), 5)