qubvel

segmentation_models.pytorch

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

segmentation_models

Segmentation models with pretrained backbones. Keras and TensorFlow Keras.

efficientnet

Implementation of EfficientNet model. Keras and TensorFlow Keras.

classification_models

Classification models trained on ImageNet. Keras.

ttach

Image Test Time Augmentation with PyTorch!

tta_wrapper

Test Time image Augmentation (TTA) wrapper for Keras model.

open-cities-challenge

First place solution for Open Cities DrivenData challenge

segmentation-models-pytorch

EfficientNet-PyTorch

A PyTorch implementation of EfficientNet

qubvel

volumentations

Library for 3D augmentations

albumentations

fast image augmentation library and easy to use wrapper around other libraries

ipyvolume

3d plotting for Python in the Jupyter notebook based on IPython widgets using WebGL

catalyst

Reproducible and fast DL & RL.

dvc-smp-test

vision

Datasets, Transforms and Models specific to Computer Vision

lightning-hydra-template

PyTorch Lightning + Hydra. A feature-rich template for rapid, scalable and reproducible ML experimentation with best practices. ⚡🔥⚡

notebooks

Notebooks using the Hugging Face libraries 🤗

omniglue

Code release for CVPR'24 submission 'OmniGlue'

transformers

🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.

accelerated_features

Implementation of XFeat (CVPR 2024). Do you need robust and fast local feature extraction? You are in the right place!

blog

Public repo for HF blog posts

catalyst_sphinx_theme

docker-python

Kaggle Python docker image

DocRes

[CVPR 2024] DocRes: A Generalist Model Toward Unifying Document Image Restoration Tasks

evaluate

🤗 Evaluate: A library for easily evaluating machine learning models and datasets.

huggingface_hub

The official Python client for the Huggingface Hub.

MONAI

AI Toolkit for Healthcare Imaging

silero-vad

Silero VAD: pre-trained enterprise-grade Voice Activity Detector

streamlit

Streamlit — The fastest way to build data apps in Python