Is there any way to export polygon coordinates of the mask of a Yolo v7 model?

97 Views Asked by At

I am training a Yolo v7 model on a dataset that I annotated, using Wong Kin Yiu's repository. The goal is to create a mask of the heart during surgery. I have had success with the model creating the mask of the heart, but as far as I am aware, I can only export the bounding box coordinates. I need to be able to export the polygon coordinates of the mask.

I tried to look through the source code and mask creating functions to find if there was a parameter I needed to export this. I could not find anything; only bounding box data.

1

There are 1 best solutions below

0
J_H On

You have a mask.

All that remains is to turn it into a convex hull. This is explained in many places, including here.