amaarora

pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more

YOLOX

YOLOX is a high-performance anchor-free YOLO, exceeding yolov3~v5 with MegEngine, ONNX, TensorRT, ncnn, and OpenVINO supported. Documentation: https://yolox.readthedocs.io/

segmentation_models.pytorch

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

EfficientNet-PyTorch

A PyTorch implementation of EfficientNet

accelerate

🚀 A simple way to launch, train, and use PyTorch models on almost any device and distributed configuration, automatic mixed precision (including fp8), and easy-to-configure FSDP and DeepSpeed support

deep_sort

Simple Online Realtime Tracking with a Deep Association Metric

FairMOT

[IJCV-2021] FairMOT: On the Fairness of Detection and Re-Identification in Multi-Object Tracking

Yolov5_StrongSORT_OSNet

Mult-object tracking and segmentation using YOLOv5 with StrongSORT and OSNet

fastpages

An easy to use blogging platform, with enhanced support for Jupyter Notebooks.

SimCLR

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

deepschool.io

Deep Learning tutorials in jupyter notebooks.

efficientdet-pytorch

A PyTorch impl of EfficientDet faithful to the original Google impl w/ ported weights

gen-efficientnet-pytorch

Pretrained EfficientNet, EfficientNet-Lite, MixNet, MobileNetV3 / V2, MNASNet A1 and B1, FBNet, Single-Path NAS

cnnimageretrieval-pytorch

CNN Image Retrieval in PyTorch: Training and evaluating CNNs for Image Retrieval in PyTorch

ECANet

Code for ECA-Net: Efficient Channel Attention for Deep Convolutional Neural Networks

COVID-CT

COVID-CT-Dataset: A CT Scan Dataset about COVID-19

fastprogress

Simple and flexible progress bar for Jupyter Notebook and console

fastcore

Python supercharged for the fastai library

Practical-Deep-Learning-for-Coders-2.0

Notebooks for the "A walk with fastai2" Study Group and Lecture Series

fastai_dev

fast.ai early development experiments

manifold_mixup

Code for reproducing Manifold Mixup results (ICML 2019)

up-detr

[TPAMI 2022 & CVPR2021 Oral] UP-DETR: Unsupervised Pre-training for Object Detection with Transformers

detecting-fake-text

Giant Language Model Test Room

pneumothorax-segmentation

SIIM-ACR Pneumothorax Segmentation first place solution

kaggle-landmark-recognition-2020-1st-place

fast_template

A template for really easy blogging with GitHub Pages

ISIC-Archive-Downloader

A script to download the ISIC Archive of lesion images

rsna-intracranial

Kaggle_RSNA2020_6th_Solution

pytorch-groupnorm

Group Normalization in PyTorch