search

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

DeepSEED: 3D Squeeze-and-Excitation Encoder-Decoder ConvNets for Pulmonary Nodule Detection
MIT License
106 stars 32 forks source link

Multi class support？ #3

Closed coolKeen closed 5 years ago

coolKeen commented 5 years ago

@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!

ymli39 commented 5 years ago

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.