Library for training deep neural nets in Pytorch.
Project description
PytorchDeepML
This library is a wrapper around pytorch and useful for solving image classification and semantic segmentation problems.
Features
-
Easy to use wrapper around pytorch so that you can focus on training and validating your model.
-
Integrates with Tensorboard to use it to monitor metrics while model trains.
-
Quickly visualize your model's predictions.
Installation
Before installing deepml, it is recommended to refer pytorch official page for torch installation.
Pypi
pip install deepml
Examples
Check out the below google colaboratory notebook examples:
- Image Regression.
- Image Classification.
- Semantic Segmentation. [Coming Soon]
Contributing
If you find a bug or have a feature request, file an issue at https://github.com/sagar-rathod/PytorchDeepML/issues
Project details
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.