TensorNet is a high-level deep learning library built on top of PyTorch.

Installation

You can use pip to install tensornet

pip install torch-tensornet

If you want to get the latest version of the code before it is released on PyPI you can install the library from GitHub

pip install git+https://github.com/shan18/TensorNet.git#egg=torch-tensornet

Documentation

Documentation is available at: https://tensornet.readthedocs.io/en/latest/

How to Use

For examples on how to use TensorNet, refer to the examples directory.

Dependencies

TensorNet has the following third-party dependencies

• torch
• torchvision
• torchsummary
• tqdm
• matplotlib
• albumentations
• opencv-python

Contact/Getting Help

If you need any help or want to report a bug, raise an issue in the repo.