There is a newer version of the record available.

Published March 11, 2021 | Version v0.2.0
Software Open

phborba/pytorch_segmentation_models_trainer: Version 0.2.0

Authors/Creators

Description

New custom callbacks:
  • ImageSegmentationResultCallback: Callback that logs the results of the training on TensorBoard and on saved files; and
  • WarmupCallback: Applies freeze weight on encoder during callback epochs and then unfreezes the weights after the warmup epochs.
Metrics added to Segmentation Model:
  • Accuracy;
  • Precision;
  • Recall; and
  • Jaccard Index (IoU).

Files

phborba/pytorch_segmentation_models_trainer-v0.2.0.zip

Files (3.8 MB)

Additional details

Related works

Is supplement to
https://github.com/phborba/pytorch_segmentation_models_trainer/tree/v0.2.0 (URL)