Mask_RCNN icon indicating copy to clipboard operation
Mask_RCNN copied to clipboard

Multi Class - Imbalance Dataset

Open avinash-218 opened this issue 1 year ago • 1 comments

I am using instance segmentation task with SFPI dataset which has imbalance data. These are the losses i could arrive at, loss :0.5265 mrcnn_bbox_loss : 0.09073 mrcnn_class_loss :0.1468 mrcnn_mask_loss:0.2024 rpn_bbox_loss:0.07144 rpn_class_loss:0.01511 val_loss:0.5339 val_mrcnn_bbox_loss:0.09229 val_mrcnn_class_loss: 0.1499 val_mrcnn_mask_loss:0.2025 val_rpn_Bbox_loss:0.07311 val_rpn_class_loss:0.0161

when visualizedm the results are very poor. mAp is also 0. Can anyone help? PS: I am using tf2

avinash-218 avatar Jun 30 '23 04:06 avinash-218

Use some Generator to Balance the data.

MDSALMANSHAMS avatar Jul 18 '23 05:07 MDSALMANSHAMS