def prepare_evolution(self, poly, extreme_point, img_init_polys, can_init_polys, img_gt_polys, can_gt_polys):
x_min, y_min = np.min(extreme_point[:, 0]), np.min(extreme_point[:, 1])
x_max, y_max = np.max(extreme_point[:, 0]), np.max(extreme_point[:, 1])
octagon = snake_coco_utils.get_octagon(extreme_point)
img_init_poly = snake_coco_utils.uniformsample(octagon, snake_config.poly_num)
can_init_poly = snake_coco_utils.img_poly_to_can_poly(img_init_poly, x_min, y_min, x_max, y_max)
img_gt_poly = snake_coco_utils.uniformsample(poly, len(poly) * snake_config.gt_poly_num)
tt_idx = np.argmin(np.power(img_gt_poly - img_init_poly[0], 2).sum(axis=1))
img_gt_poly = np.roll(img_gt_poly, -tt_idx, axis=0)[::len(poly)]
can_gt_poly = snake_coco_utils.img_poly_to_can_poly(img_gt_poly, x_min, y_min, x_max, y_max)
img_init_polys.append(img_init_poly)
can_init_polys.append(can_init_poly)
img_gt_polys.append(img_gt_poly)
can_gt_polys.append(can_gt_poly)
def prepare_init(self, box, extreme_point, i_it_4pys, c_it_4pys, i_gt_4pys, c_gt_4pys, h, w):
x_min, y_min = np.min(extreme_point[:, 0]), np.min(extreme_point[:, 1])
x_max, y_max = np.max(extreme_point[:, 0]), np.max(extreme_point[:, 1])
img_init_poly = snake_coco_utils.get_init(box)
img_init_poly = snake_coco_utils.uniformsample(img_init_poly, snake_config.init_poly_num)
can_init_poly = snake_coco_utils.img_poly_to_can_poly(img_init_poly, x_min, y_min, x_max, y_max)
img_gt_poly = extreme_point
can_gt_poly = snake_coco_utils.img_poly_to_can_poly(img_gt_poly, x_min, y_min, x_max, y_max)
i_it_4pys.append(img_init_poly)
c_it_4pys.append(can_init_poly)
i_gt_4pys.append(img_gt_poly)
c_gt_4pys.append(can_gt_poly)