imagenet · GitHub Topics

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT)...

Python 34.45 k

16 小时前

HumanSignal / labelImg

#计算机科学#LabelImg is now part of the Label Studio community. The popular image annotation tool created by Tzutalin is no longer actively being developed, but you can check out Label Studio, the open source dat...

Python annotations 工具 imagenet 深度学习 detection recognition image-classification

Python 23.84 k

1 年前

microsoft / Swin-Transformer

This is an official implementation for "Swin Transformer: Hierarchical Vision Transformer using Shifted Windows".

swin-transformer image-classification object-detection semantic-segmentation imagenet mscoco ade20k mask-rcnn

Python 14.89 k

1 年前

cvat-ai / cvat

#计算机科学#CVAT 是一个领先的工业级用于机器学习的图片、视频标注工具。

Python 13.89 k

2 天前

qubvel-org / segmentation_models.pytorch

Semantic segmentation models with 500+ pretrained convolutional and transformer-based backbones.

segmentation 图像处理 pspnet unet unet-pytorch PyTorch fpn models imagenet semantic-segmentation image-segmentation segmentation-models deeplabv3 deeplab-v3-plus pretrained-weights 机器视觉 transformers

Python 10.55 k

3 天前

Cadene / pretrained-models.pytorch

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

imagenet resnet resnext pretrained PyTorch inception

Python 9.09 k

3 年前

lukemelas / EfficientNet-PyTorch

A PyTorch implementation of EfficientNet

efficientnet-pytorch imagenet feature-extraction pretrained-models

Python 8.12 k

3 年前

tensorlayer / TensorLayer

#计算机科学#Deep Learning and Reinforcement Learning Library for Scientists and Engineers

tensorlayer 深度学习 Tensorflow 神经网络 reinforcement-learning 人工智能 Generative Adversarial Network a3c dqn object-detection 聊天机器人 Python imagenet Google

Python 7.36 k

2 年前

ml5js / ml5-library

#计算机科学#Friendly machine learning for the web! 🤖

lstm 深度学习 imagenet 机器学习 JavaScript 神经网络 p5xjs p5js

JavaScript 6.55 k

8 个月前

huawei-noah / Efficient-AI-Backbones

Efficient AI Backbones including GhostNet, TNT and MLP, developed by Huawei Noah's Ark Lab.

convolutional-neural-networks efficient-inference imagenet model-compression Tensorflow PyTorch ghostnet transformer pretrained-models vision-transformer

Python 4.24 k

3 个月前

tinyvision / DAMO-YOLO

#计算机科学#DAMO-YOLO: a fast and accurate object detection method with some new techs, including NAS backbones, efficient RepGFPN, ZeroHead, AlignedOTA, and distillation enhancement.

深度学习 nas object-detection onnx PyTorch tensorrt yolo yolov5 imagenet damo-yolo

Python 3.07 k

1 年前

osmr / imgclsmob

#计算机科学#Sandbox for training deep learning networks

机器学习深度学习 mxnet gluon PyTorch classification imagenet 神经网络 image-classification chainer Keras Tensorflow pretrained-models cifar segmentation tensorflow2 semantic-segmentation human-pose-estimation 3d-face-reconstruction

Python 3.01 k

9 个月前

mit-han-lab / efficientvit

Efficient vision foundation models for high-resolution generation and perception.

high-resolution imagenet efficientvit segment-anything segmentation vision-transformer deep-compression-autoencoder efficient-diffusion-model

Python 2.91 k

2 个月前

qubvel / efficientnet

#计算机科学#Implementation of EfficientNet model. Keras and TensorFlow Keras.

classification imagenet efficientnet pretrained-models image-classification efficient nasnetmobile mobilenet 深度学习

Python 2.09 k

1 年前

apple / ml-cvnets

#计算机科学#CVNets: A library for training computer vision networks

ade20k classification 机器视觉深度学习 detection imagenet 机器学习 mscoco pascal-voc PyTorch segmentation

Python 1.88 k

2 年前

TsingZ0 / PFLlib

Master Federated Learning in 2 Hours—Run It on Your PC!

federated-learning personalization PyTorch Python distributed-computing differential-privacy 隐私 imagenet Internet of things attack privacy-protection

Python 1.78 k

5 天前

bearpaw / pytorch-classification

Classification with PyTorch.

classification cifar10 cifar100 imagenet PyTorch resnet resnext densenet

Python 1.73 k

1 年前

google-research / big_transfer

#计算机科学#Official repository for the "Big Transfer (BiT): General Visual Representation Learning" paper.

深度学习 convolutional-neural-networks imagenet tensorflow2 jax PyTorch transfer-learning

Python 1.53 k

1 年前

yoshitomo-matsubara / torchdistill

#自然语言处理#A coding-free framework built on PyTorch for reproducible deep learning studies. PyTorch Ecosystem. 🏆25 knowledge distillation methods presented at CVPR, ICLR, ECCV, NeurIPS, ICCV, etc are implemente...

knowledge-distillation PyTorch image-classification imagenet object-detection coco semantic-segmentation cifar10 cifar100 colab-notebook google-colab pascal-voc 自然语言处理 transformer glue text-classification

Python 1.52 k

1 个月前

hysts / pytorch_image_classification

PyTorch implementation of image classification models for CIFAR-10/CIFAR-100/MNIST/FashionMNIST/Kuzushiji-MNIST/ImageNet

PyTorch 机器视觉 cifar10 fashion-mnist imagenet

Python 1.41 k

4 年前