pysegcnn.core.models

Neural networks for semantic image segmentation.

License

Copyright (c) 2020 Daniel Frisinghelli

This source code is licensed under the GNU General Public License v3.

See the LICENSE file in the repository’s root directory.

Classes

Network()

Generic Network class.

SupportedLossFunctions(value)

Names and corresponding classes of the tested loss functions.

SupportedModels(value)

Names and corresponding classes of the implemented models.

SupportedOptimizers(value)

Names and corresponding classes of the tested optimizers.

UNet(in_channels, nclasses, filters, skip, …)

A PyTorch implementation of `U-Net`_.