Amiiney

pytorch

Tensors and Dynamic neural networks in Python with strong GPU acceleration

pytorch-image-models

The largest collection of PyTorch image encoders / backbones. Including train, eval, inference, export scripts, and pretrained weights -- ResNet, ResNeXT, EfficientNet, NFNet, Vision Transformer (ViT), MobileNetV4, MobileNet-V3 & V2, RegNet, DPN, CSPNet, Swin Transformer, MaxViT, CoAtNet, ConvNeXt, and more

pytorch-grad-cam

Advanced AI Explainability for computer vision. Support for CNNs, Vision Transformers, Classification, Object detection, Segmentation, Image similarity and more.

segmentation_models.pytorch

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

streamlit-webrtc

Real-time video and audio processing on Streamlit

traingenerator

🧙 A web app to generate template code for machine learning

classification_models

Classification models trained on ImageNet. Keras.

dfdc_deepfake_challenge

A prize winning solution for DFDC challenge

tph-yolov5

DnaFeaturesViewer

:eye: Python library to plot DNA sequence features (e.g. from Genbank files)

Next-ViT

bioinfokit

Bioinformatics data analysis and visualization toolkit

cvpr2020-plant-pathology

Robust-and-efficient-post-processing-for-video-object-detection

awesome-mixed-sample-data-augmentation

A collection of awesome things about mixed sample data augmentation

Kaggle-PANDA-1st-place-solution

1st place solution for the Kaggle PANDA Challenge

Cell-DETR

Official and maintained implementation of the paper "Attention-Based Transformers for Instance Segmentation of Cells in Microstructures" [BIBM 2020].

UMA-MOT

A Unified Object Motion and Affinity Model for Online Multi-Object Tracking (CVPR2020)

pymetric

pytorch metric learning tools and pycls

seq-nms

Implementation of the seq-nms post-processing algorithm for video object detection

rendezvous

A transformer-inspired neural network for surgical action triplet recognition from laparoscopic videos.

kaggle_sartorius

Solution for the Sartorius - Cell Instance Segmentation competition

MODNet

Docker version API for MODNet-model Human Matting

ivtmetrics

A Python evaluation metrics package for surgical action triplet recognition

timmFasterRcnn

Adds timm pretrained backbone to pytorch's FasterRcnn model

ranzcr-clip

Kaggle | 22nd place solution for RANZCR CLiP - Catheter and Line Position Challenge.

LG_Plant_Disease_Diagnosis

DACON AI Competition for Plant Disease Classification

DistillableVisionTransformer

Easy usage of distillation tokens with vision transformers. Includes models pretrained on ImageNet.

Streamlit

Streamlit examples

gbr-starfish-kaggle

TensorFlow - Help Protect the Great Barrier Reef. Detect crown-of-thorns starfish in underwater image data