Exemplo n.º 1
0
destination += "tests/"

if os.path.exists(destination):
    destination += "{}".format(int(time.time())) + "/"
    os.makedirs(destination)
else:
    os.makedirs(destination)

# Save snake parameters to file for reference
filename = destination + "param.txt"
f = open(filename, "w+")
f.write(name)
f.close()

processed_image = Contours.setup_image(image, scale, sigma)
scaled_line = Contours.scale_line(line, scale)

snake = active_contour(snake=scaled_line,
                       image=processed_image,
                       alpha=alpha,
                       beta=beta,
                       w_line=w_line,
                       w_edge=w_edge,
                       gamma=gamma,
                       bc=ends,
                       convergence=convergence)

failed = Contours.failed_snake(line, Contours.scale_line(snake, 1 / scale),
                               relaxation, difference)
Contours.save_snakes([snake],
                     image,