Convolution Open Source Projects

Machine learning, in numpy

MNN: A blazing-fast, lightweight inference engine battle-tested by Alibaba, powering high-performance on-device LLMs and Edge AI.

A technical report on convolution arithmetic in the context of deep learning

CUDA Templates and Python DSLs for High-Performance Linear Algebra

Imaging is a simple image processing package for Go

A Swift library that uses the Accelerate framework to provide high-performance functions for matrix math, digital signal processing, and image manipulation.

Machine Learning and Computer Vision Engineer - Technical Interview Questions

This repo contains the source code in my personal column (https://zhuanlan.zhihu.com/zhaoyeyu), implemented using Python 3.6. Including Natural Language Processing and Computer Vision projects, such as text generation, machine translation, deep convolution GAN and other actual combat code.

校招、秋招、春招、实习好项目！带你从零实现一个高性能的深度学习推理库，支持大模型 llama2 、Unet、Yolov5、Resnet等模型的推理。Implement a high-performance deep learning inference library step by step

Semantic Segmentation Suite in TensorFlow. Implement, train, and test new Semantic Segmentation models easily!

Graph Neural Networks with Keras and Tensorflow 2.

Torch implementation of ResNet from http://arxiv.org/abs/1512.03385 and training scripts

Spatial Sparse Convolution Library

Pytorch 3D U-Net Convolution Neural Network (CNN) designed for medical image segmentation

Submanifold sparse convolutional networks

Speedy Wavenet generation using dynamic programming :zap:

Vulkan/CUDA/HIP/OpenCL/Level Zero/Metal Fast Fourier Transform library

ResNet model in TensorFlow

An implementation of the system-wide JamesDSP audio processing engine for non-rooted Android devices

Deformable ConvNets V2 (DCNv2) in PyTorch

[MICRO'23, MLSys'22] TorchSparse: Efficient Training and Inference Framework for Sparse Convolution on GPUs.

Deep Learning in Haskell

Deep Learning Zero to All - Pytorch

A New Padding Scheme: Partial Convolution based Padding

A PyTorch implementation of "Capsule Graph Neural Network" (ICLR 2019).

Fully convolutional deep neural network to remove transparent overlays from images

船长关于机器学习、计算机视觉和工程技术的总结和分享

Replication of simple CV Projects including attention, classification, detection, keypoint detection, etc.

Theory of digital signal processing (DSP): signals, filtration (IIR, FIR, CIC, MAF), transforms (FFT, DFT, Hilbert, Z-transform) etc.

Arbitrary style transfer using TensorFlow.js

DetectoRS: Detecting Objects with Recursive Feature Pyramid and Switchable Atrous Convolution

[Unofficial] PyTorch implementation of "Conformer: Convolution-augmented Transformer for Speech Recognition" (INTERSPEECH 2020)

Source code for "On the Relationship between Self-Attention and Convolutional Layers"

Image processing and manipulation in JavaScript

an implementation of Video Frame Interpolation via Adaptive Separable Convolution using PyTorch

Library for specialized dense and sparse matrix operations, and deep learning primitives.

Material used for Deep Learning related workshops for Machine Learning Tokyo (MLT)

PyTorch implementation of Deformable Convolution

Unofficial implementation of "Image Inpainting for Irregular Holes Using Partial Convolutions". Try at: www.fixmyphoto.ai

Audio DSP effects build on Android system framework layer. This is a repository contains a pack of high quality DSP algorithms specialized for audio processing.

Kernel Point Convolution implemented in PyTorch

ACNet: Strengthening the Kernel Skeletons for Powerful CNN via Asymmetric Convolution Blocks

FastFCN: Rethinking Dilated Convolution in the Backbone for Semantic Segmentation.

常用的语义分割架构结构综述以及代码复现

This project accelerates CNN computation with the help of FPGA, for more than 50x speed-up compared with CPU.

Dilated Convolution for Semantic Image Segmentation

Kernel Point Convolutions

dabnn is an accelerated binary neural networks inference framework for mobile platform

PyTorch implementation of Deformable ConvNets v2 (Modulated Deformable Convolution)

CondenseNet: Light weighted CNN for mobile devices

E(2)-Equivariant CNNs Library for Pytorch

Code for the book Deep Learning From Scratch, from O'Reilly September 2019

Winograd minimal convolution algorithm generator for convolutional neural networks.

Understanding Convolution for Semantic Segmentation

Unofficial pytorch implementation of 'Image Inpainting for Irregular Holes Using Partial Convolutions' [Liu+, ECCV2018]

Building Convolutional Neural Networks From Scratch using NumPy

Pytorch!!!Pytorch!!!Pytorch!!! Dynamic Convolution: Attention over Convolution Kernels (CVPR-2020)

Pytorch implementation of newly added convolution

PyTorch implementation of Glow