This Project introduces BB-UNet: Unet with bounding box prior.
The most important file of the repository is the model_2.py that contains the BB-UNet model and the BB-CONV layer. The BB-UNet can hence be plugged into any other learning framework and adapted to any dataset.
Other files includes the loss function scripts , the data processing script and the curve plotting evolutions but are specific the SegTHOR and heart implementations.