Convolutional Neural Networks Python Packages

swift-f0

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

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).

neurite

Neural networks toolbox focused on medical image analysis

xnns

Neural network interatomic potentials for chemistry and materials science

antspynet

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

toothless

Radio galaxy classifier for python

hypercube-wtf

Frozen hypercube reservoir for static high-dimensional fields (spectra, sensors, images). Short synthetic orbit, HypercubeCNN on the end state only. C++23 + Python.

biapy

Open source Python library for building bioimage analysis pipelines

neuralnetlib

🧠 A flexible machine & deep learning framework built from scratch using only NumPy

axondeepseg

Axon/Myelin segmentation using Deep Learning

networks

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

fastmri

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

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.

dropblock

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

keras-hexagdly

Keras 3 hexagonal CNN layers (HexagDLy port) for hexagonally-sampled data — IACT/Cherenkov cameras, etc. — Support for hls4ml in progress

unet

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

continual-inference

A Python library for Continual Inference Networks in PyTorch

minkowskiengine

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

efficientword-net

OneShot Learning-based hotword detection.

miscnn

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

orthoseg

Train your own neural networks to segment orthophotos.

cosmic-conn

Cosmic-CoNN: A Cosmic Ray Detection Deep-Learning Framework, Dataset, and Toolkit

pydtnn

PyDTNN - Python Distributed Training of Neural Networks