def ExitPrediction(self, roi, xExitShift, yExitShift, xRef, yRef):
img = self.ModExit.predict(roi)
img = cv2.resize(img, (xExitShift * 2, yExitShift * 2))
x, y = met.getCenter(img, xExitShift * 2, yExitShift * 2)
x = int(x)
y = int(y)
return x + xRef - xExitShift, y + yRef - yExitShift
def OpticDiscPrediction(self):
img = self.Mod.predict(self.currentImg)
# img = moil.stackImageChannels(img)
# resizing prediction
w, h, c = moil.getWidthHeightChannels(self.currentImg)
img = cv2.resize(img, (w, h))
# getting coords
x, y = met.getCenter(img, w, h)
x = int(x)
y = int(y)
return x, y, img