Catalyst

catalyst-team/catalyst

High-level utils for PyTorch DL & RL research. It was developed with a focus on reproducibility, fast experimentation and code/ideas reusing. Being able to research/develop something new, rather than write another regular train loop.

  • deep-learning
  • reinforcement-learning
  • machine-learning
  • computer-vision
  • pytorch
  • python
  • distributed-computing
  • infrastructure

More in Python

Serpent.AISerpentAI/SerpentAI Archived

Serpent.AI is a game agent framework that allows you to turn any video game you own into a sandbox to develop AI and machine learning experiments. For both researchers and hobbyists.

Deep learningVisionRL
7k archived

RLlibray-project/ray

RLlib is an industry level, highly scalable RL library for tf and torch, based on Ray. It's used by companies like Amazon and Microsoft to solve real-world decision making problems at scale.

Deep learningRL
44.1k updated today

timmrwightman/pytorch-image-models

PyTorch image models, scripts, pretrained weights -- ResNet, ResNeXT, EfficientNet, EfficientNetV2, NFNet, Vision Transformer, MixNet, MobileNet-V3/V2, RegNet, DPN, CSPNet, and more.

Deep learningVision
37.2k updated 4d ago

detectron2facebookresearch/detectron2

FAIR's next-generation research platform for object detection and segmentation. It is a ground-up rewrite of the previous version, Detectron, and is powered by the PyTorch deep learning framework.

Deep learningVision
34.9k updated 4d ago

ML-From-Scratcheriklindernoren/ML-From-Scratch

Implementations of Machine Learning models from scratch in Python with a focus on transparency. Aims to showcase the nuts and bolts of ML in an accessible way.

GeneralDeep learningRL
33.1k updated 3y ago

Detectronfacebookresearch/Detectron Archived

FAIR's software system that implements state-of-the-art object detection algorithms, including Mask R-CNN. It is written in Python and powered by the Caffe2 deep learning framework.

Deep learningVision
26.3k archived
O

OpenFacecmusatyalab/openface

Free and open source face recognition with deep neural networks.

Deep learningVision
15.4k updated 10d ago

albumentationsalbu/albumentations Archived

А fast and framework agnostic image augmentation library that implements a diverse set of augmentation techniques. Supports classification, segmentation, detection out of the box. Was used to win a number of Deep Learning competitions at Kaggle, Topcoder and those that were a part of the CVPR workshops.

Deep learningVision
15.3k archived
All 443 in Python
Esc

Type to search every entry. ↑↓ to move, Enter to open.