I have a set of polygons for each image. Those polygons consist of four $x$ and $y$ coordinates. For each image, I need to extract the ones of interest. This could be formulated as an Image Segmentation task where, for example I want to extract the objects of interest, here: cars.
But since I already get the polygons through a different part of my pipeline I would like to create a simpler machine learning model. The input will not be the image but only the coordinates of the polygons. In this model each sample should consist of multiple polygons (those can vary in number) and the model should output the ones of interest.
In my mind, I formulated the problem as follows:
- The polygons are the features. Problem: Samples will have varying number of features.
- The output will consist of the indices of the "features" (polygons) I am interested in.
First, I created a decision tree and classified each coordinate as $0$ (not interested in) or $1$ (of interest). But, by doing this, I don't consider the other coordinates that belong to the image. The information of the surrounding is lost.
Does someone have an idea of how to model this problem without using Image Segmentation?