tp030ny/nn-additional-losses

A collection of losses not part of pytorch standard library particularly useful for segmentation task

NN Additional Losses

A collection of losses particularly for segmentation task. Please use technical documentation for a reference to API manual

Getting Started

Prerequisites

You need to install pytorch in order for this work

pip install torch

Installing

This is not yet available on Pypi so you need to clone the repo in order to install

Git clone

git clone https://github.com/shayansiddiqui/nn-additional-losses

Then go to the cloned directory and install locally (Be sure of of environment where you are installing)

pip install -e .

Now the package is installed and ready to be used across all pytorch projects

Authors

• Shayan Ahmad Siddiqui - shayansiddiqui
• Abhijit Guha Roy - abhi4ssj

License

This project is licensed under the MIT License - see the LICENSE.md file for details