All Categories → Machine Learning → imagenet

Top 152 imagenet open source projects

DAWNBench: An End-to-End Deep Learning Benchmark and Competition

✭ 254

python deeplearning imagenet training inference cifar10 squad

Reference ImageNet implementation of SelecSLS CNN architecture proposed in the SIGGRAPH 2020 paper "XNect: Real-time Multi-Person 3D Motion Capture with a Single RGB Camera". The repository also includes code for pruning the model based on implicit sparsity emerging from adaptive gradient descent methods, as detailed in the CVPR 2019 paper "On implicit filter level sparsity in Convolutional Neural Networks".

✭ 251

python deep-learning pytorch cnn imagenet pruning efficient

Mobilenetv3 Pytorch

Implementing Searching for MobileNetV3 paper using Pytorch

✭ 243

python pytorch imagenet

Pyramidnet Pytorch

A PyTorch implementation for PyramidNets (Deep Pyramidal Residual Networks, https://arxiv.org/abs/1610.02915)

✭ 234

python pytorch resnet imagenet

Pyconv

Pyramidal Convolution: Rethinking Convolutional Neural Networks for Visual Recognition (https://arxiv.org/pdf/2006.11538.pdf)

✭ 231

python deep-learning machine-learning pytorch computer-vision neural-network artificial-intelligence deep-neural-networks convolutional-neural-networks cnn imagenet image-recognition

Fusenet

Deep fusion project of deeply-fused nets, and the study on the connection to ensembling

✭ 230

jupyter-notebook resnet mxnet imagenet fusion

Octconv.pytorch

PyTorch implementation of Octave Convolution with pre-trained Oct-ResNet and Oct-MobileNet models

✭ 229

python deep-neural-networks resnet imagenet mobilenet

Triplet Attention

Official PyTorch Implementation for "Rotate to Attend: Convolutional Triplet Attention Module." [WACV 2021]

✭ 222

jupyter-notebook deep-learning computer-vision convolutional-neural-networks detection paper attention-mechanism imagenet arxiv

Moga

MoGA: Searching Beyond MobileNetV3

✭ 220

python pytorch imagenet

Mini Imagenet Tools

Tools for generating mini-ImageNet dataset and processing batches

✭ 209

python dataset imagenet meta-learning

Labelimg

🖍️ LabelImg is a graphical image annotation tool and label object bounding boxes in images

✭ 16,088

python shell Makefile deep-learning tools detection image-classification annotations imagenet recognition

Sequential Imagenet Dataloader

A plug-in replacement for DataLoader to load ImageNet disk-sequentially in PyTorch.

✭ 198

python deep-learning pytorch imagenet

Atomnas

Code for ICLR 2020 paper 'AtomNAS: Fine-Grained End-to-End Neural Architecture Search'

✭ 197

python distributed imagenet neural-architecture-search

Pytorch Cpp

PyTorch C++ inference with LibTorch

✭ 194

cpp pytorch demo opencv imagenet inference

Torchdistill

PyTorch-based modular, configuration-driven framework for knowledge distillation. 🏆18 methods including SOTA are implemented so far. 🎁 Trained models, training logs and configurations are available for ensuring the reproducibiliy.

✭ 177

python pytorch object-detection semantic-segmentation image-classification imagenet cifar10 coco

Imgclsmob

Sandbox for training deep learning networks

Senet Caffe

A Caffe Re-Implementation of SENet

✭ 169

caffe resnet imagenet

Iresnet

Improved Residual Networks (https://arxiv.org/pdf/2004.04989.pdf)

✭ 163

python deep-learning machine-learning pytorch computer-vision artificial-intelligence deep-neural-networks convolutional-neural-networks neural-networks cnn image-classification resnet imagenet image-recognition

Imagenet

TensorFlow implementation of AlexNet and its training and testing on ImageNet ILSVRC 2012 dataset

✭ 155

python deep-learning tensorflow convolutional-neural-networks imagenet

Models Comparison.pytorch

Code for the paper Benchmark Analysis of Representative Deep Neural Network Architectures

✭ 148

python pytorch deep-neural-networks convolutional-neural-networks imagenet

Alexnet

implement AlexNet with C / convolutional nerual network / machine learning / computer vision

✭ 147

c computer-vision image-classification imagenet

Shufflenet V2 Tensorflow

A lightweight convolutional neural network

✭ 145

jupyter-notebook tensorflow classification imagenet

Efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

✭ 1,920

python shell Dockerfile deep-learning classification image-classification imagenet pretrained-models mobilenet efficient nasnetmobile efficientnet

Tensorrt Utils

⚡ Useful scripts when using TensorRT

✭ 136

python docker gpu optimization nvidia imagenet

Mobilenetworks

Keras implementation of Mobile Networks

✭ 132

python tensorflow keras neural-network imagenet

Aognet

Code for CVPR 2019 paper: " Learning Deep Compositional Grammatical Architectures for Visual Recognition"

✭ 132

python deep-learning computer-vision deep-neural-networks image-classification mxnet imagenet cifar10

Vision4j Collection

Collection of computer vision models, ready to be included in a JVM project

✭ 132

java computer-vision grpc classification segmentation semantic-segmentation jvm imagenet

Pytorch Imagenet Cifar Coco Voc Training

Training examples and results for ImageNet(ILSVRC2012)/CIFAR100/COCO2017/VOC2007+VOC2012 datasets.Image Classification/Object Detection.Include ResNet/EfficientNet/VovNet/DarkNet/RegNet/RetinaNet/FCOS/CenterNet/YOLOv3.

✭ 130

python pytorch classification detection resnet apex imagenet yolov3 darknet coco

Regnet

Pytorch implementation of network design paradigm described in the paper "Designing Network Design Spaces"

✭ 129

python deep-learning pytorch classification image-classification imagenet

Imagenet

This implements training of popular model architectures, such as AlexNet, ResNet and VGG on the ImageNet dataset(Now we supported alexnet, vgg, resnet, squeezenet, densenet)

✭ 126

python pytorch resnet imagenet densenet vgg

Pyramidnet

Torch implementation of the paper "Deep Pyramidal Residual Networks" (https://arxiv.org/abs/1610.02915).

✭ 121

lua deep-learning resnet imagenet torch7

Labelbox

Labelbox is the fastest way to annotate data to build and ship computer vision applications.

Ir Net

This project is the PyTorch implementation of our accepted CVPR 2020 paper : forward and backward information retention for accurate binary neural networks.

✭ 119

python pytorch resnet imagenet

Petridishnn

Code for the neural architecture search methods contained in the paper Efficient Forward Neural Architecture Search

✭ 112

python deep-learning image-classification automl imagenet neural-architecture-search cifar10 deep-learning-algorithms

Moco

Unofficial implementation with pytorch DistributedDataParallel for "MoCo: Momentum Contrast for Unsupervised Visual Representation Learning"

✭ 112

python pytorch unsupervised-learning imagenet

Imagenetv2

A new test set for ImageNet

✭ 109

jupyter-notebook dataset imagenet

Resnet Imagenet Caffe

train resnet on imagenet from scratch with caffe

✭ 105

shell caffe resnet imagenet

Ghostnet

CV backbones including GhostNet, TinyNet and TNT, developed by Huawei Noah's Ark Lab.

✭ 1,744

python pytorch tensorflow convolutional-neural-networks imagenet cvpr model-compression transformer pretrained-models efficient-inference ghostnet vision-transformer

Pnasnet.tf

TensorFlow implementation of PNASNet-5 on ImageNet

✭ 102

python deep-learning tensorflow automl imagenet neural-architecture-search

Vovnet.pytorch

A pytorch implementation of VoVNet

✭ 101

python imagenet

Hbonet

[ICCV 2019] Harmonious Bottleneck on Two Orthogonal Dimensions

✭ 94

python pytorch imagenet pretrained-models mobilenetv2

Pytorch Classification

Classification with PyTorch.

✭ 1,268

python pytorch classification resnet imagenet densenet cifar10 resnext

Tf Mobilenet V2

Mobilenet V2(Inverted Residual) Implementation & Trained Weights Using Tensorflow

✭ 85

python tensorflow convolutional-neural-networks google cnn deeplearning image-classification imagenet mobilenet

Caffe Model

Caffe models (including classification, detection and segmentation) and deploy files for famouse networks

✭ 1,258

python classification segmentation detection caffe resnet imagenet faster-rcnn inception resnext

Mobilenet Caffe

Caffe Implementation of Google's MobileNets (v1 and v2)

✭ 1,217

python caffe imagenet mobilenet mobilenetv2

Chainer Pspnet

PSPNet in Chainer

✭ 76

python deep-learning convolutional-neural-networks semantic-segmentation imagenet chainer cityscapes pascal-voc

Artos

Adaptive Real-Time Object Detection System with HOG and CNN Features

✭ 64

python machine-learning computer-vision gui image-processing object-detection imagenet machinelearning machine-learning-library

Pneumonia Detection From Chest X Ray Images With Deep Learning

Detecting Pneumonia in Chest X-ray Images using Convolutional Neural Network and Pretrained Models

✭ 64

jupyter-notebook tensorflow keras convolutional-neural-networks imagenet medical-imaging image-recognition keras-tensorflow inception

Tf Hrnet

tensorflow implementation for "High-Resolution Representations for Labeling Pixels and Regions"

✭ 61

python tensorflow imagenet

One Pixel Attack Keras

Keras implementation of "One pixel attack for fooling deep neural networks" using differential evolution on Cifar10 and ImageNet

✭ 1,097

jupyter-notebook deep-learning machine-learning tensorflow nlp keras neural-network image-processing cnn imagenet cifar10

Big transfer

Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

✭ 1,096

python deep-learning pytorch convolutional-neural-networks transfer-learning imagenet

Biglittlenet

Official repository for Big-Little Net

✭ 57

python pytorch computer-vision speech-recognition imagenet image-recognition

Imagenet

Trial on kaggle imagenet object localization by yolo v3 in google cloud

✭ 56

python jupyter-notebook image-classification yolo google-cloud imagenet yolov3

Divide And Co Training

[Paper 2020] Towards Better Accuracy-efficiency Trade-offs: Divide and Co-training. Plus, an image classification toolbox includes ResNet, Wide-ResNet, ResNeXt, ResNeSt, ResNeXSt, SENet, Shake-Shake, DenseNet, PyramidNet, and EfficientNet.

✭ 54

python image-classification imagenet

Pretrained Models.pytorch

Pretrained ConvNets for pytorch: NASNet, ResNeXt, ResNet, InceptionV4, InceptionResnetV2, Xception, DPN, etc.

✭ 8,318

python lua pytorch resnet imagenet inception resnext pretrained

Segmentationcpp

A c++ trainable semantic segmentation library based on libtorch (pytorch c++). Backbone: ResNet, ResNext. Architecture: FPN, U-Net, PAN, LinkNet, PSPNet, DeepLab-V3, DeepLab-V3+ by now.

✭ 49

cpp pytorch neural-network semantic-segmentation resnet imagenet image-segmentation models unet resnext

Imagenet resnet tensorflow2.0

Train ResNet on ImageNet in Tensorflow 2.0; ResNet 在ImageNet上完整训练代码

✭ 42

python resnet imagenet

Constrained attention filter

(ECCV 2020) Tensorflow implementation of A Generic Visualization Approach for Convolutional Neural Networks

✭ 36

python python3 tensorflow deep-neural-networks convolutional-neural-networks classification imagenet autonomous-driving autonomous-vehicles

Dcgan

Porting pytorch dcgan on FloydHub