Rust library for image segmentation using Burn.
The overall architecture and concept was inspired from qubvel-org/segmentation_models.pytorch.
For each encoder/decoder,
- EfficientNet encoder is based on lukemelas/EfficientNet-PyTorch.
- ResNet encoder was copied from tracel-ai/models.
- Unet++ decoder is based on qubvel-org/segmentation_models.pytorch.
This library is distributed under MIT license, as in LICENSE file.
ResNet encoder code partially belongs to Burn-rs/Models Contributors. Please see tracel-ai/models LICENSE-MIT.