ymli39/DeepSEED-3D-ConvNets-for-Pulmonary-Nodule-Detection

Multi class support?

coolKeen opened this issue · 1 comments

@ymli39 Hi, thanks for sharing your code and paper. I wonder whether it supports multi-class detection when I use luna_detector (not only detect pulmonary nodule). If not, how can I modify the code to support multi-class detection?
Thanks in advance!

The original code does not support multi-class detection. In addition to pass the bounding boxes as gt, you could also pass the label and modify the loss function.