def generateInferior2MeanMatrix(self):
if self.GEN_MEAN_INF_ZONE_MAT:
vietImcp = self.loadImageCol()
vietslice = vietImcp[:, 550:600]
carver = SeamMarker(img=vietImcp)
emap = carver.calc_energy(vietslice)
normalized = normalizeImageVals(emap.copy())
# vietEmap = ImageOps.grayscale(Image.fromarray(emap))
funcs = SeamFuncsAI()
zones = funcs.getInferior2MeanZones(normalized)
np.save(self.mean_inf_zone_mat_path, zones)
def test_seammarker_seam_ai_goRightDown(self):
r, c = 2, 1
compr, compc = 3, 2
funcs = SeamFuncsAI()
outr, outc = funcs.goRightDown(r, c)
self.assertEqual((outr, outc), (compr, compc))
def test_seammarker_seam_ai_goLeftUp(self):
r, c = 2, 1
compr, compc = 1, 0
funcs = SeamFuncsAI()
outr, outc = funcs.goLeftUp(r, c)
self.assertEqual((outr, outc), (compr, compc))
def test_seammarker_minimum_seam_ai(self):
vietImcp = self.loadImageCol()
vietslice = vietImcp[:, 550:600]
carver = SeamMarker(img=vietImcp)
emap = carver.calc_energy(vietslice)
funcs = SeamFuncsAI()