Trending AI reposUpdated daily from GitHub

72.8% MobileNetV2 1.0 model on ImageNet and a spectrum of pre-trained MobileNetV2 models

Random Erasing Data Augmentation. Experiments on CIFAR10, CIFAR100 and Fashion-MNIST

NeuPy is a Tensorflow based python library for prototyping and building neural networks

Vision-Centric BEV Perception: A Survey

An open-source PyTorch code for crowd counting

Fast CUDA implementation of (differentiable) soft dynamic time warping for PyTorch

Dataset of images of trash; Torch-based CNN for garbage image classification

Visual Question Answering in Pytorch

A curated list of papers, theses, datasets, and tools related to the application of Machine Learning for Software Engineering

[CVPR 2021 Best Student Paper Honorable Mention, Oral] Official PyTorch code for ClipBERT, an efficient framework for end-to-end learning on image-text and video-text tasks.

An Open-Source Library for Training Binarized Neural Networks

Implementation of TimeSformer from Facebook AI, a pure attention-based solution for video classification

RL starter files in order to immediately train, visualize and evaluate an agent without writing any line of code

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

Inference Code for Polygon-RNN++ (CVPR 2018)

Code for paper "Computation Offloading Optimization for UAV-assisted Mobile Edge Computing: A Deep Deterministic Policy Gradient Approach"

A 3D electromagnetic FDTD simulator written in Python with optional GPU support

Differentiable signal processing on the sphere for PyTorch

A modular, open and non-proprietary toolkit for core robotic functionalities by harnessing deep learning

FFA-Net: Feature Fusion Attention Network for Single Image Dehazing

Revisiting Pre-trained Models for Chinese Natural Language Processing (MacBERT)

Pytorch实现的流式与非流式的自动语音识别框架，同时兼容在线和离线识别，目前支持Conformer、Squeezeformer、DeepSpeech2模型，支持多种数据增强方法。

TensorFlow and Deep Learning Tutorials

Resources on Music Generation with Deep Learning

Simple implementation of OpenAI CLIP model in PyTorch.

LightNet: Light-weight Networks for Semantic Image Segmentation (Cityscapes and Mapillary Vistas Dataset)

A concise but complete implementation of CLIP with various experimental improvements from recent papers

The official implementation of MARS: Unleashing the Power of Variance Reduction for Training Large Models

Image Restoration with Mean-Reverting Stochastic Differential Equations, ICML 2023. Winning solution of the NTIRE 2023 Image Shadow Removal Challenge.

This repository contains the source codes for the paper "AtlasNet: A Papier-Mâché Approach to Learning 3D Surface Generation ". The network is able to synthesize a mesh (point cloud + connectivity) from a low-resolution point cloud, or from an image.

Open-source Monocular Python HawkEye for Tennis

Support PointRend, Fast_SCNN, HRNet, Deeplabv3_plus(xception, resnet, mobilenet), ContextNet, FPENet, DABNet, EdaNet, ENet, Espnetv2, RefineNet, UNet, DANet, HRNet, DFANet, HardNet, LedNet, OCNet, EncNet, DuNet, CGNet, CCNet, BiSeNet, PSPNet, ICNet, FCN, deeplab)

Fine-tune pretrained Convolutional Neural Networks with PyTorch

Code for "Neural Controlled Differential Equations for Irregular Time Series" (Neurips 2020 Spotlight)

ATen: A TENsor library for C++11

End-To-End Molecular Dynamics (MD) Engine using PyTorch

Learning to Cluster Faces (CVPR 2019, CVPR 2020)

Real-Time Semantic Segmentation in Mobile device

AI 算法岗简历模板

[NeurIPS 2020] MCUNet: Tiny Deep Learning on IoT Devices; [NeurIPS 2021] MCUNetV2: Memory-Efficient Patch-based Inference for Tiny Deep Learning

CVPR 论文收集，包含但不限于2022、2021、2020、2019、2018、2017文章

Reimplementation of World-Models (Ha and Schmidhuber 2018) in pytorch

A Keras+TensorFlow Implementation of the Transformer: Attention Is All You Need

FreeVC: Towards High-Quality Text-Free One-Shot Voice Conversion

Template repository to build PyTorch projects from source on any version of PyTorch/CUDA/cuDNN.

Personal deep learning study notes and tutorial-style notebooks

Google Cloud Platform Vertex AI end-to-end workflows for machine learning operations

Fully Convlutional Neural Networks for state-of-the-art time series classification