Skip to content

Ability to use image volumes with no boxes for training as negatives #6791

@AceMcAwesome77

Description

@AceMcAwesome77

Hi, I think it would be an important improvement of the MONAI detector models to have the ability to include "negative" image volumes, i.e. image volumes with no pathology bounding boxes, in the training dataset. The current RetinaNet 3D code for example requires each image volume to have at least 1 bounding box. There are many real-world examples where we would need negatives to train an accurate model. For example, if training a model to detect blood clots in a particular artery, you would also need to include image volumes showing that artery without any clot - otherwise, if you only trained on images of clotted arteries at that location, the detector would just classify that artery location as positive whether it was clotted or not.

I made some minor changes to the Retinanet 3D code that accomplished this, but there may be a more efficient way than mine so I did not branch any code. I have expanded on this issue in more detail here:

Project-MONAI/tutorials#1292

Thanks!

Metadata

Metadata

Assignees

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions