MIC-DKFZ / medicaldetectiontoolkit

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.3k stars 297 forks source link

Plot legend missing when using --resume_to_checkpoint #110

Open rtgunti opened 4 years ago

rtgunti commented 4 years ago

Hello! Thanks for the amazing repo! Below is the training plot on a custom dataset when trained for 5 epochs. train_pre When I try to resume the training from the last checkpoint, the legend is missing from the updated training plot train_resume