search

matterport / Mask_RCNN

Mask R-CNN for object detection and instance segmentation on Keras and TensorFlow
Other
24.55k stars 11.69k forks source link

Multi Class - Imbalance Dataset #2966

Open avinash-218 opened 1 year ago

avinash-218 commented 1 year ago

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

MDSALMANSHAMS commented 1 year ago

Use some Generator to Balance the data.