AutoML for image augmentation. AutoAlbument uses the Faster AutoAugment algorithm to find optimal augmentation policies. Documentation - https://albumentations.ai/docs/autoalbument/

✭ 99

python deep-learning machine-learning pytorch computer-vision automl automated-machine-learning augmentation

Tta wrapper

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

✭ 98

python tensorflow keras image-classification image-segmentation augmentation

Face landmark dnn

Face Landmark Detector based on Mobilenet V1

✭ 92

python keras cnn dataset coreml mobilenet augmentation

Fast Autoaugment

Official Implementation of 'Fast AutoAugment' in PyTorch.

✭ 1,297

python deep-learning pytorch computer-vision convolutional-neural-networks cnn distributed image-classification automl automated-machine-learning augmentation

Albumentations

Fast image augmentation library and an easy-to-use wrapper around other libraries. Documentation: https://albumentations.ai/docs/ Paper about the library: https://www.mdpi.com/2078-2489/11/2/125

✭ 9,353

python deep-learning machine-learning image-processing object-detection segmentation detection image-classification image-segmentation augmentation image-augmentation fast-augmentations

Pytorch Toolbelt

PyTorch extensions for fast R&D prototyping and Kaggle farming

✭ 942

python deep-learning machine-learning pytorch image-processing object-detection segmentation image-classification pipeline kaggle image-segmentation augmentation

Segmentation

Catalyst.Segmentation

✭ 27

python deep-learning machine-learning docker pytorch image-processing segmentation pipeline image-segmentation reproducibility augmentation

Cutmix Pytorch

Official Pytorch implementation of CutMix regularizer

✭ 754

python transfer-learning augmentation

Torchio

Medical image preprocessing and augmentation toolkit for deep learning

✭ 708

python deep-learning machine-learning pytorch computer-vision convolutional-neural-networks cnn segmentation medical-imaging data-augmentation augmentation

Sketch Code

Keras model to generate HTML code from hand-drawn website mockups. Implements an image captioning architecture to drawn source images.

✭ 4,544

python shell deep-learning tensorflow keras image-processing augmentation

Augmentor

Image augmentation library in Python for machine learning.

✭ 4,594

python deep-learning machine-learning neural-networks augmentation

Pba

Efficient Learning of Augmentation Policy Schedules

✭ 461

python jupyter-notebook deep-learning machine-learning tensorflow data-science artificial-intelligence convolutional-neural-networks image-classification automl data-augmentation automated-machine-learning augmentation

Ttach

Image Test Time Augmentation with PyTorch!

✭ 455

python deep-learning pytorch computer-vision classification segmentation augmentation

Caer

High-performance Vision library in Python. Scale your research, not boilerplate.

✭ 452

python deep-learning machine-learning computer-vision data-science neural-network artificial-intelligence image-processing ai opencv gpu cuda segmentation image-classification image-segmentation video-processing vision augmentation type-checking

Audiomentations

A Python library for audio data augmentation. Inspired by albumentations. Useful for machine learning.

✭ 439

python deep-learning machine-learning audio music sound dsp data-augmentation augmentation

Instaboost

Code for ICCV2019 paper "InstaBoost: Boosting Instance Segmentation Via Probability Map Guided Copy-Pasting"

✭ 368

python segmentation augmentation

Deltapy

DeltaPy - Tabular Data Augmentation (by @firmai)

✭ 344

jupyter-notebook machine-learning data-science time-series finance data-augmentation feature-extraction feature-engineering tabular-data augmentation

Pytorch Randaugment

Unofficial PyTorch Reimplementation of RandAugment.

✭ 323

python deep-learning pytorch computer-vision convolutional-neural-networks classification imagenet augmentation

Toolbox

various cv tools, such as label tools, data augmentation, label conversion, etc.

✭ 279

jupyter-notebook coco augmentation

pydiogment

📣 Python library for audio augmentation

✭ 64

python TeX Makefile audio machine-learning deep-learning sound sound-processing audio-processing augmentation audio-effects

Data Augmentation Zoo for Object Detection

Includes: Learning data augmentation strategies for object detection | GridMask data augmentation | Augmentation for small object detection in Numpy. Use RetinaNet with ResNet-18 to test these methods on VOC and KITTI.

✭ 126

python augmentation objectdetection gridmask smallobjectdetection

rmpe dataset server

Realtime Multi-Person Pose Estimation data server. Used as a training and validation data provider in training process.

✭ 14

C++computer-vision realtime human-pose-estimation augmentation pose-estimation human-behavior-understanding cvpr-2017

DataAugmentationNMT

Data Augmentation for Neural Machine Translation

✭ 26

lua perl python shell neural-machine-translation language-model augmentation

augmenty

Augmenty is an augmentation library based on spaCy for augmenting texts.

✭ 101

python nlp natural-language-processing text-classification spacy augmentation training-data nlproc spacy-nlp spacy-extension text-augmentation

IR-GAN

Augmenting Room Impulse Response

✭ 21

matlab python Jupyter Notebook javascript c HTML automatic-speech-recognition augmentation room-impulse-response synthetic-data

timber-ruby

🌲 Great Ruby logging made easy.

✭ 155

ruby rails metadata application sinatra events hanami log logging context structured-logging timber timber-ruby contextual-data augmentation lograge

SQUAD2.Q-Augmented-Dataset

Augmented version of SQUAD 2.0 for Questions

✭ 31

python google translation deep-learning neural-network dataset question-answering squad bert augmentation

discolight

discolight is a robust, flexible and infinitely hackable library for generating image augmentations ✨

✭ 25

python shell machine-learning computer-vision deep-learning image-processing bounding-boxes image-augmentation augmentation image-transformations

torch-pitch-shift

Pitch-shift audio clips quickly with PyTorch (CUDA supported)! Additional utilities for searching efficient transformations are included.

✭ 70

python torch pytorch sound-processing augmentation pitch-shift gpu-support torchaudio audio-augmentation

classification

Catalyst.Classification

✭ 35

shell python Dockerfile Makefile docker machine-learning deep-learning pipeline image-processing pytorch classification image-classification reproducibility catalyst augmentation focal-loss classification-pipeline

volumentations

Library for 3D augmentations

✭ 111

python transformations 3d augmentation 3d-augmentation

imgcrop

Simple image augmentation library focusing on random geometric cropping

✭ 27

python deep-neural-networks deep-learning image-processing segmentation convolutional-neural-networks augmentation

CAP augmentation

Cut and paste augmentation for object detection and instance segmentation

✭ 93

Jupyter Notebook self-driving-car segmentation object-detection semantic-segmentation augmentation instance-segmentation

textaugment

TextAugment: Text Augmentation Library

✭ 280

python nlp natural-language-processing word2vec wordnet hacktoberfest synonym augmentation mixup augmentation-methods low-resouce-language nlp-augmentation

mix3d

Mix3D: Out-of-Context Data Augmentation for 3D Scenes (3DV 2021 Oral)

✭ 183

python shell point-cloud semantic-segmentation 3d augmentation 3dvision

FAST-RIR

This is the official implementation of our neural-network-based fast diffuse room impulse response generator (FAST-RIR) for generating room impulse responses (RIRs) for a given acoustic environment.

✭ 90

python shell deep-learning neural-network speech impulse-response generative-adversarial-network automatic-speech-recognition rir augmentation acoustics room-impulse-response synthetic-data conditional-generation diffuse-scattering implicit-neural-representation

deep utils

An open-source toolkit which is full of handy functions, including the most used models and utilities for deep-learning practitioners!

✭ 73

python nlp machine-learning computer-vision deep-learning tensorflow utils pytorch coco face-recognition face-detection object-detection augmentation senet vggface2 cutmix yolov5 modelcheckpoint

1-45 of 45 augmentation projects