v0.6.1

Added

• PFAN architecture
• sigmoid option in inference script

Fixed

• model_name check in create_model

v0.5.4

Added

• Listing models
• Weights for pretrained models in README.md
• Inference entry point
• Saliency maps visualization utils

v0.4.0

Added

• Added PGNet, U2Net models
• Added PGNet and MultiBCE losses
• Added SaliencyDataset and SaliencyTransform utils
• Added train experiment entry point
• Added utils to train experiment
• Added pytorch_lightning datamodule

Fixed

• Correct casting between tensors in different computations