-
Notifications
You must be signed in to change notification settings - Fork 1.2k
Description
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:
Thanks!