Question Regarding Polygon Input Support in GeoLayoutLM

I really appreciate your work on GeoLayoutLM. I would like to ask if there is any way to support polygon input with four points instead of only rectangular bounding boxes . Is it possible to modify the model’s 2D segment box embedding or add an extension so that it can directly accept polygon coordinate inputs ? Any guidance or recommended approach would be greatly appreciated.

Thank you for your time and assistance