femshima / segmentation-models-burn

Rust library for image segmentation using Burn.

MIT License

3 stars 0 forks source link

burn computer-vision deep-learning image-segmentation models rust segmentation-models semantic-segmentation

readme

segmentation-models-burn

Rust library for image segmentation using Burn.

Acknowledgements

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.

License

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.