Convolutional Neural Networks Python Packages

antialiased-cnns

pip install antialiased-cnns to improve stability and accuracy

saliency

Framework-agnostic implementation for state-of-the-art saliency methods (XRAI, BlurIG, SmoothGrad, and more).

swift-f0

Fast and accurate fundamental frequency (F0) detector using convolutional neural networks

neurite

Neural networks toolbox focused on medical image analysis

fastmri

A large-scale dataset of both raw MRI measurements and clinical MRI images.

unet

"pip install unet": PyTorch Implementation of 1D, 2D and 3D U-Net architecture.

dropblock

Implementation of DropBlock: A regularization method for convolutional networks in PyTorch.

axondeepseg

Axon/Myelin segmentation using Deep Learning

easytorch

EasyTorch is a research-oriented pytorch prototyping framework with a straightforward learning curve. It is highly robust and contains almost everything needed to perform any state-of-the-art experiments.

pydtnn

PyDTNN - Python Distributed Training of Neural Networks

orthoseg

Package to make it easier to segment orthophotos.

miscnn

A framework for Medical Image Segmentation with Convolutional Neural Networks and Deep Learning

antspynet

Pre-trained models and utilities for deep learning on medical images in Python (Keras/TensorFlow)

minkowskiengine

Minkowski Engine is an auto-diff neural network library for high-dimensional sparse tensors

neuralnetlib

A flexible deep learning framework built from scratch using only NumPy

networks

Library which can be used to build feed forward NN, Convolutional Nets, Linear Regression, and Logistic Regression Models.

biapy

Open source Python library for building bioimage analysis pipelines

yolox-backbone

yolox_backbone is a deep-learning library and is a collection of YOLOX Backbone models.

continual-inference

A Python library for Continual Inference Networks in PyTorch

pixellib

Visit PixelLib's official documentation https://pixellib.readthedocs.io/en/latest/

zensols-deeplearn

General deep learning tools

pupyl

🧿 Pupyl is a really fast image search library which you can index your own (millions of) images and find similar images in milliseconds.

cad-recognition

An image kernel is a small matrix used to apply effects like the ones you might find in Photoshop or Gimp, such as blurring, sharpening, outlining or embossing.

moai-mdk

moai: Accelerated, Flexible, Modular, Reproducible, Insightful AI