How to get GFL esk loss with segmentation masks

open-mmlab / mmdetection

OpenMMLab Detection Toolbox and Benchmark

https://mmdetection.readthedocs.io

Apache License 2.0

29.49k stars 9.44k forks source link

How to get GFL esk loss with segmentation masks #11245

Open GeorgePearse opened 11 months ago

GeorgePearse commented 11 months ago

I have a heavily imbalanced dataset, GFL learns it really well, but I'm now in need of segmentation output and cascade mask rcnn seems to be struggling.

I'm a little limited in my choice because I want to be able to export to ONNX, so a change in loss would be ideal, but several other issues seemed to have had little success with this approach

GeorgePearse commented 11 months ago

This is a more general question wrt. what are my options to improve performance on an imbalanced dataset for a mask_rcnn type model. Not specific to loss functions which I've seen discussed elsewhere.