Stars
A latent text-to-image diffusion model
🔊 Text-Prompted Generative Audio Model
A game theoretic approach to explain the output of any machine learning model.
A simple screen parsing tool towards pure vision based GUI agent
Audiocraft is a library for audio processing and generation with deep learning. It features the state-of-the-art EnCodec audio compressor / tokenizer, along with MusicGen, a simple and controllable…
Grounded SAM: Marrying Grounding DINO with Segment Anything & Stable Diffusion & Recognize Anything - Automatically Detect , Segment and Generate Anything
Foundational Models for State-of-the-Art Speech and Text Translation
Using Low-rank adaptation to quickly fine-tune diffusion models.
Taming Transformers for High-Resolution Image Synthesis
YOLOv6: a single-stage object detection framework dedicated to industrial applications.
Repo for the Deep Reinforcement Learning Nanodegree program
PyTorch implementation of AnimeGANv2
2-2000x faster ML algos, 50% less memory usage, works on all hardware - new and old.
Kernl lets you run PyTorch transformer models several times faster on GPU with a single line of code, and is designed to be easily hackable.
Google Colaboratory Notebooks and Repositories (by @firmai)
Official PyTorch repo for JoJoGAN: One Shot Face Stylization
A smaller subset of 10 easily classified classes from Imagenet, and a little more French
Implementation of the paper "DeepLSD: Line Segment Detection and Refinement with Deep Image Gradients"
Attempting to replicate "A Deep Reinforcement Learning Framework for the Financial Portfolio Management Problem" https://arxiv.org/abs/1706.10059 (and an openai gym environment)
Detectorch - detectron for PyTorch
Cool experiments at the intersection of Computer Vision and Sports ⚽🏃
Reinforcement Learning for Portfolio Management
PyTorch implementation of DeepDream algorithm (Mordvintsev et al.). Additionally I've included playground.py to help you better understand basic concepts behind the algo.
A journey into Convolutional Neural Network visualization
Reproduce research from paper "Predicting the direction of stock market prices using random forest"
Lightweight gaze estimation with PyTorch.