segmentation_models. Segmentation models with pretrained backbones. Keras and TensorFlow Keras.
4.9kefficientnet. Implementation of EfficientNet model. Keras and TensorFlow Keras.
2.1kclassification_models. Classification models trained on ImageNet. Keras.
1.2kttach. Image Test Time Augmentation with PyTorch!
1ktransformers-notebooks. Inference and fine-tuning examples for vision models from 🤗 Transformers
169tta_wrapper. Test Time image Augmentation (TTA) wrapper for Keras model.
112residual_attention_network. Keras implementation of Residual Attention Network
109rt-pose. Real-time pose estimation pipeline with 🤗 Transformers
72open-cities-challenge. First place solution for Open Cities DrivenData challenge
48keras_telegram_callback. Telegram-bot callback for your Keras model
31awesome-semantic-segmentation. :metal: awesome-semantic-segmentation
12awesome-satellite-imagery-competitions. List of machine learning competitions for satellite imagery and remote sensing.
9albumentations. fast image augmentation library and easy to use wrapper around other libraries
6segmentation-models-pytorch.
5qubvel.
5keras-applications. Reference implementations of popular deep learning models.
5resnet_152. Pretrained on ImageNet ResNet-152 model in Keras
5EfficientNet-PyTorch. A PyTorch implementation of EfficientNet
5keras_lr_finder. Plots the change of the loss function of a Keras model when the learning rate is exponentially increasing.
5assets.
5template. Template for data science project based on Docker and Jupyter server with Keras/Pytorch
4transformers. 🤗 Transformers: State-of-the-art Machine Learning for Pytorch, TensorFlow, and JAX.
4volumentations. Library for 3D augmentations
3MMdnn. MMdnn is a set of tools to help users inter-operate among different deep learning frameworks. E.g. model conversion and visualization. Convert models between Caffe, Keras, MXNet, Tensorflow, CNTK, PyTorch Onnx and CoreML.
3train_imagenet. Python
3ipyvolume. 3d plotting for Python in the Jupyter notebook based on IPython widgets using WebGL
2dvc-smp-test. Python
2catalyst. Reproducible and fast DL & RL.
2vision. Datasets, Transforms and Models specific to Computer Vision
2notebooks. Notebooks using the Hugging Face libraries 🤗
2omniglue. Code release for CVPR'24 submission 'OmniGlue'
2blog. Public repo for HF blog posts
1accelerate. 🚀 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
1MONAI. AI Toolkit for Healthcare Imaging
1tpu. Reference models and tools for Cloud TPUs.
1DocRes. [CVPR 2024] DocRes: A Generalist Model Toward Unifying Document Image Restoration Tasks
1accelerated_features. Implementation of XFeat (CVPR 2024). Do you need robust and fast local feature extraction? You are in the right place!
1huggingface_hub. The official Python client for the Huggingface Hub.
1lightning-hydra-template. PyTorch Lightning + Hydra. A feature-rich template for rapid, scalable and reproducible ML experimentation with best practices. ⚡🔥⚡
1keras. Deep Learning for humans
1sentiment_analysis. MVideo hackathon test task. User`s review rating prediction
1