All Categories → Machine Learning → imagenet

Top 152 imagenet open source projects

simpleAICV-pytorch-ImageNet-COCO-training

SimpleAICV:pytorch training example on ImageNet(ILSVRC2012)/COCO2017/VOC2007+2012 datasets.Include ResNet/DarkNet/RetinaNet/FCOS/CenterNet/TTFNet/YOLOv3/YOLOv4/YOLOv5/YOLOX.

✭ 276

python shell pytorch classification imagenet coco resnet object-detection darknet cifar100 voc distillation retinanet yolov3 fcos centernet yolov4 yolov5 ttfnet ilsvrc2012 yolox mosaic-augment

image-classification

A collection of SOTA Image Classification Models in PyTorch

✭ 70

python Jupyter Notebook shell deep-learning cnn pytorch transformer imagenet image-classification mlp quantization object-recognition sota vision-transformer mlp-mixer

TF-NAS

TF-NAS: Rethinking Three Search Freedoms of Latency-Constrained Differentiable Neural Architecture Search (ECCV2020)

✭ 66

python shell imagenet nas neural-architecture-search tf-nas searched-architectures search-freedoms latency-constrained

super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library

✭ 429

Jupyter Notebook python computer-vision deep-learning neural-network pytorch imagenet image-classification object-detection transfer-learning pretrained-models semantic-segmentation pretrained-weights

PyTorch-Model-Compare

Compare neural networks by their feature similarity

✭ 119

python deep-learning transformers pytorch pip feature-extraction neural-networks imagenet cka torch-cka

etiketai

Etiketai is an online tool designed to label images, useful for training AI models

✭ 63

javascript CSS machine-learning recognition tools ai deep-learning preact images nextjs detection annotations labels artificial-intelligence yolo imagenet image-classification hacktoberfest labelimg

ShapeTextureDebiasedTraining

Code and models for the paper Shape-Texture Debiased Neural Network Training (ICLR 2021)

✭ 95

python shape paper texture pytorch imagenet

SKNet-PyTorch

Nearly Perfect & Easily Understandable PyTorch Implementation of SKNet

✭ 62

python pytorch imagenet cvpr imagenet-classifier cvpr2019 selective-kernel-networks sknet

datumaro

Dataset Management Framework, a Python library and a CLI tool to build, analyze and manage Computer Vision datasets.

✭ 274

python HTML SCSS javascript CSS Batchfile computer-vision deep-learning dataset neural-networks yolo imagenet coco format-converter datasets pascal-voc openvino-toolkit

imagenet-autoencoder

Autoencoder trained on ImageNet Using Torch 7

✭ 18

lua imagenet autoencoder torch7

Stochastic-Quantization

Training Low-bits DNNs with Stochastic Quantization

✭ 70

Jupyter Notebook C++python Cuda CMake Protocol Buffer deep-neural-networks caffe imagenet cifar quantization

SharpPeleeNet

ImageNet pre-trained SharpPeleeNet can be used in real-time Semantic Segmentation/Objects Detection

✭ 13

python pytorch imagenet object-detection semantic-segmentation pytorch-implmention peleenet sharp-peleenet lightnet

DetectionMetrics

Tool to evaluate deep-learning detection and segmentation models, and to create datasets

✭ 66

C++python CMake c Slice shell deep-learning tensorflow keras imagenet coco object-detection darknet pascal-voc object-segmentation

efficientnetv2.pytorch

PyTorch implementation of EfficientNetV2 family

✭ 366

python imagenet pytorch-implementation efficientnet efficientnetv2 icml2021

BottleneckTransformers

Bottleneck Transformers for Visual Recognition

✭ 231

python transformers imagenet image-classification image-recognition cifar10 bottleneck-transformers

ImageModels

ImageNet model implemented using the Keras Functional API

✭ 63

Jupyter Notebook python keras image-processing mkdocs imagenet imagenet-classifier mkdocs-material

img classification deep learning

No description or website provided.

✭ 19

Jupyter Notebook react search django image word2vec imagenet image-classification image-search inception ann knn googlenet image-retrieval image-similarity image-search-engine

PyTorch-LMDB

Scripts to work with LMDB + PyTorch for Imagenet training

✭ 49

lmdb pytorch imagenet resnet dataloader resnet50

SAN

[ECCV 2020] Scale Adaptive Network: Learning to Learn Parameterized Classification Networks for Scalable Input Images

✭ 41

python pytorch classification imagenet resnet knowledge-distillation scale-invariance meta-learning mobilenetv2 efficient-neural-networks eccv2020

ghostnet.pytorch

73.6% GhostNet 1.0x pre-trained model on ImageNet

✭ 90

python pytorch imagenet pretrained-models reproduction mobilenetv3 ghostnet

Parametric-Contrastive-Learning

Parametric Contrastive Learning (ICCV2021)

✭ 155

python Jupyter Notebook shell Dockerfile Batchfile Makefile pytorch supervised-learning imagenet image-classification class-imbalance imbalanced-data imbalanced-learning contrastive-learning supervised-contrastive-learning long-tailed-recognition iccv2021 parametric-contrastive-learning

Xception-caffe

Xception implemented with caffe

✭ 45

caffe imagenet xception-caffe xception

Skin Lesions Classification DCNNs

Transfer Learning with DCNNs (DenseNet, Inception V3, Inception-ResNet V2, VGG16) for skin lesions classification

✭ 47

Jupyter Notebook computer-vision imagenet image-classification ensemble-learning densenet transfer-learning vgg16 inceptionv3 inception-resnet biomedical-image-processing fine-tuning skin-cancer skin-lesion-classification

cisip-FIRe

Fast Image Retrieval (FIRe) is an open source project to promote image retrieval research. It implements most of the major binary hashing methods to date, together with different popular backbone networks and public datasets.

TokenLabeling

Pytorch implementation of "All Tokens Matter: Token Labeling for Training Better Vision Transformers"

✭ 385

Jupyter Notebook python shell pytorch transformer vision imagenet segmentation lv-vit

py-faster-rcnn-imagenet

Train faster rcnn on imagine dataset, related blog post: https://andrewliao11.github.io/object/detection/2016/07/23/detection/

✭ 133

imagenet faster-rcnn

BAKE

Self-distillation with Batch Knowledge Ensembling Improves ImageNet Classification

✭ 79

python shell classification imagenet knowledge-distillation

cozmo-tensorflow

🤖 Cozmo the Robot recognizes objects with TensorFlow

✭ 61

python deep-learning tensorflow floyd-cli floydhub imagenet transfer-learning cozmo-sdk cozmo

head-network-distillation

[IEEE Access] "Head Network Distillation: Splitting Distilled Deep Neural Networks for Resource-constrained Edge Computing Systems" and [ACM MobiCom HotEdgeVideo 2019] "Distilled Split Deep Neural Networks for Edge-assisted Real-time Systems"

✭ 27

Jupyter Notebook python pytorch imagenet image-classification knowledge-distillation split-computing

alexnet

custom implementation alexnet with tensorflow

✭ 21

python shell tensorflow cnn imagenet alexnet convolutional-neural-networks

nested-transformer

Nested Hierarchical Transformer https://arxiv.org/pdf/2105.12723.pdf

✭ 174

Jupyter Notebook python shell transformer vision imagenet

EffcientNetV2

EfficientNetV2 implementation using PyTorch

✭ 94

python shell training pytorch imagenet efficientnetv2

121-152 of 152 imagenet projects

first

‹