Problem: A data synthesizer generates images and labels. The goal is to train a model with at most 4.5 million trainable parameters which determines whether each image has a star and, if so, finds a rotated bounding box that bounds the star.
More precisely, the labels contain the following five numbers, which your model should predict:
- the x and y coordinates of the center
- yaw
- width and height.
If there is no star, the label consists of 5 np.nans. The height of the star is always noticeably larger than its width, and the yaw points in one of the height directions. The yaw is always in the interval [0, 2 * pi), oriented counter-clockwise and with zero corresponding to the upward direction.
Important Points
- ResNet18 inspired Network for rotated object detection using Modulated Rotation Loss
- Architecture: ResNet18 with classification and regression heads
- Loss Function: Modulated Rotation Loss
- Model Summary: modelSummary.txt
- Final Score: score.txt
- Final Model weights: model.pickle
- Version requirements: requirements.txt
- Configuration for teh model: config.yaml
Method
- Generated a fixed number of images (defined in the config file) for train and validation dataloader with high noise and rotated objects. The probability of star being present in any image is set to 0.8
- Normalize the values such that the range of the labels is [0, 1]
- Add a classification label for every image based on the values of the generated labels
- The images are passed through a ResNet18 inspired model with seprate classification and regression heads
- Implement modulated loss function, to bridge the discontinuity in the values of rotation angle (yaw)
Opportunities to Improve:
- Add a more complex network (Upsampling from the bottleneck or FPN) for the regression head
- Improvise the loss function to take into account the imbalance between the classes