The Medical Detection Toolkit contains 2D + 3D implementations of prevalent object detectors such as Mask R-CNN, Retina Net, Retina U-Net, as well as a training and inference framework focused on dealing with medical images.
Apache License 2.0
1.31k
stars
297
forks
source link
Plot legend missing when --resume_to_checkpoint #111
Fixes #110
Cause : Legend is added to the plot only for the first epoch. Fix: Changed the condition to add the legend for 'starting_epoch' instead.