A neural network toolkit.
Project description
# pytorch_modules
## Introduction
A neural network toolkit built on pytorch/opencv/numpy that includes neural network layers, modules, loss functions, optimizers, data loaders, data augmentation, etc.
## Features
Advanced neural network modules, such as EfficientNet, ResNet, SENet, Xception, DenseNet, FocalLoss, AdaboundW
Ultra-efficient dataloader that allows you to take full advantage of GPU
High performance and multifunctional data augmentation(See [woodsgao/image_augments](https://github.com/woodsgao/image_augments))
## Installation
sudo pip3 install pytorch_modules
## Usage
### pytorch_modules.nn
This module contains a variety of neural network layers, modules and loss functions.
import torch from pytorch_modules.nn import ResBlock
# NCHW tensor inputs = torch.ones([8, 8, 224, 224]) block = ResBlock(8, 16) outputs = block(inputs)
### pytorch_modules.augments
See [woodsgao/image_augments](https://github.com/woodsgao/image_augments) for more details.
### pytorch_modules.backbones
This module includes a series of modified backbone networks, such as EfficientNet, ResNet, SENet, Xception, DenseNet.
import torch from pytorch_modules.backbones import ResNet
# NCHW tensor inputs = torch.ones([8, 8, 224, 224]) model = ResNet(32) outputs = model(inputs)
### pytorch_modules.datasets
This module includes a series of dataset classes integrated from pytorch_modules.datasets.BasicDataset which is integrated from torch.utils.data.Dataset . The loading method of pytorch_modules.datasets.BasicDataset is modified to cache data with LMDB to speed up data loading. This allows your gpu to be fully used for model training without spending a lot of time on data loading and data augmentation. Please see the corresponding repository for detailed usage.
pytorch_modules.datasets.ClassificationDataset > [woodsgao/pytorch_classification](https://github.com/woodsgao/pytorch_classification)
pytorch_modules.datasets.SegmentationDataset > [woodsgao/pytorch_segmentation](https://github.com/woodsgao/pytorch_segmentation)
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.
Source Distribution
Built Distribution
Hashes for pytorch_modules-0.2.4-py2.py3-none-any.whl
Algorithm | Hash digest | |
---|---|---|
SHA256 | 388f930a7fb09b3952d9143ee5b1e722b63cb0e2cbb3e485f4b0d5e110e0c971 |
|
MD5 | b745245aceafd9b9c4822441decc4eab |
|
BLAKE2b-256 | cd04e6769f5ee04e1cc833d02fd97a1ec03db352a4ddc5354c990d3f5f7a693a |