MultiChexNet is a model that is able to classify, detect and segment chest related disease in one forward step.
You Can use this repo to train your independent model, whether it is classification,detection or segmenter only. You can find an example in the usage_example.py
- Navigate to either the Classifier.py, Detector.py or segmenter.py and fill in the make model function
- Make an external file where you make an instance of the encoder and one of the following according to your task (classifier,detector, segmenter)
- Call the
ModelBlock.add_head(encoder, [classifier])
to add your model head to the encoder