qubvel

segmentation_models.pytorch

Segmentation models with pretrained backbones. PyTorch.

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

keras_telegram_callback

Telegram-bot callback for your Keras model

segmentation-models-pytorch

EfficientNet-PyTorch

A PyTorch implementation of EfficientNet

keras_lr_finder

Plots the change of the loss function of a Keras model when the learning rate is exponentially increasing.

keras-applications

Reference implementations of popular deep learning models.

qubvel

template

Template for data science project based on Docker and Jupyter server with Keras/Pytorch

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

MMdnn

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.

train_imagenet

dvc-smp-test

vision

Datasets, Transforms and Models specific to Computer Vision

catalyst

Reproducible and fast DL & RL.

keras

Deep Learning for humans

lightning-hydra-template

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

tpu

Reference models and tools for Cloud TPUs.

catalyst_sphinx_theme

docker-python

Kaggle Python docker image

instance-metric

overpass-api-python-wrapper

Python bindings for the OpenStreetMap Overpass API

streamlit

Streamlit — The fastest way to build data apps in Python