search

JDSobek / MedYOLO

A 3D bounding box detection model for medical data.
GNU Affero General Public License v3.0
25 stars 6 forks source link

weights for lung nodule #9

Closed Sere1nz closed 6 months ago

Sere1nz commented 6 months ago

Hi, Great paper! I am interested in med3d detection. Could you provide trained weight for LIDC or luna16(lung nodule)? Thank you very much!

JDSobek commented 6 months ago

Unfortunately no. I did try to train on LIDC several times, but the model never showed any signs of detecting the nodules. My impression is that lung nodules are too small for the model to detect. I've seen recent success on fairly small objects, but nothing that small.

I'm not very familiar with lung nodules and there may be some tricks to make lung nodule detection work that I don't know about, so I may try again in the future, but in the meantime nnDetection has pretrained weights for LIDC and luna16 so I'd suggest their model over this one for lung nodule detection.