search

Project-MONAI / model-zoo

MONAI Model Zoo that hosts models in the MONAI Bundle format.
Apache License 2.0
204 stars 69 forks source link

MONAI nodule detection model training and handling zero nodule datasets #681

Open DSRajesh opened 1 month ago

DSRajesh commented 1 month ago

Hello Thank you for the MONAI team. I have been training a nodule detection model using image

Among my datasets I have some which have 0 nodules in them. And i think i get a error log as shown below for such cases

image I have applied the correction mentioned there

i.e boxes = torch.reshape ( boxes, [ 0, 4 ] ) as follows:

image

but i still get the same error log as shown above. May i know the reason for this and the solution for the same please. Also I have used datasets that have 0 nodules and a nnDetection (https://github.com/MIC-DKFZ/nnDetection) model trains well under such situations.

Rajesh D S