Awesome Open Source

Programming Languages

Search results for dataset pytorch

1,516 search results found

Pytorch Cyclegan And Pix2pix ⭐ 20,036

Image-to-Image Translation in PyTorch

Insightface ⭐ 18,230

State-of-the-art 2D and 3D Face Analysis Project

Datasets ⭐ 17,197

🤗 The largest hub of ready-to-use datasets for ML models with fast, easy-to-use and efficient data manipulation tools

Vision ⭐ 14,575

Datasets, Transforms and Models specific to Computer Vision

Cvat ⭐ 10,136

Annotate better with CVAT, the industry-leading data engine for machine learning. Used and trusted by teams at any scale, for data of any scale.

Deeplake ⭐ 6,901

Database for AI. Store Vectors, Images, Texts, Videos, etc. Use with LLMs/LangChain. Store, query, version, & visualize any AI data. Stream data in real-time to PyTorch/TensorFlow. https://activeloop.ai

Techniques ⭐ 6,889

Techniques for deep learning with satellite & aerial imagery

🤖 💬 Deep learning for Text to Speech (Discussion forum: https://discourse.mozilla.org/c/tts)

Transformers Tutorials ⭐ 5,671

This repository contains demos I made with the Transformers library by HuggingFace.

Yet Another Efficientdet Pytorch ⭐ 4,892

The pytorch re-implement of the official efficientdet with SOTA performance in real time and pretrained weights.

Latex Ocr ⭐ 4,801

pix2tex: Using a ViT to convert images of equations into LaTeX code.

Yolact ⭐ 4,793

A simple, fully convolutional model for real-time instance segmentation.

Nnunet ⭐ 4,100

Mmdetection3d ⭐ 4,088

OpenMMLab's next-generation platform for general 3D object detection.

Nerf Pytorch ⭐ 3,981

A PyTorch implementation of NeRF (Neural Radiance Fields) that reproduces the results.

Openpcdet ⭐ 3,838

OpenPCDet Toolbox for LiDAR-based 3D Object Detection.

Deep Person Reid ⭐ 3,813

Torchreid: Deep learning person re-identification in PyTorch.

Models, data loaders and abstractions for language processing, powered by PyTorch

中文语言理解测评基准 Chinese Language Understanding Evaluation Benchmark: datasets, baselines, pre-trained models, corpus and leaderboard

Stargan V2 ⭐ 3,185

StarGAN v2 - Official PyTorch Implementation (CVPR 2020)

Perceptualsimilarity ⭐ 3,011

LPIPS metric. pip install lpips

Super Slomo ⭐ 2,873

PyTorch implementation of Super SloMo by Jiang et al.

[CVPR2020] Adversarial Latent Autoencoders

Recbole ⭐ 2,842

A unified, comprehensive and efficient recommendation library

Edge Connect ⭐ 2,404

EdgeConnect: Structure Guided Image Inpainting using Edge Prediction, ICCV 2019 https://arxiv.org/abs/1901.00212

Benchmarking Gnns ⭐ 2,196

Repository for benchmarking graph neural networks

Image_segmentation ⭐ 2,186

Pytorch implementation of U-Net, R2U-Net, Attention U-Net, and Attention R2U-Net.

Pytorch Nlp ⭐ 2,180

Basic Utilities for PyTorch Natural Language Processing (NLP)

Imbalanced Dataset Sampler ⭐ 2,073

A (PyTorch) imbalanced dataset sampler for oversampling low frequent classes and undersampling high frequent ones.

Objectron ⭐ 1,958

Objectron is a dataset of short, object-centric video clips. In addition, the videos also contain AR session metadata including camera poses, sparse point-clouds and planes. In each video, the camera moves around and above the object and captures it from different views. Each object is annotated with a 3D bounding box. The 3D bounding box describes the object’s position, orientation, and dimensions. The dataset contains about 15K annotated video clips and 4M annotated images in the following cat

Deepvoice3_pytorch ⭐ 1,851

PyTorch implementation of convolutional neural networks-based text-to-speech synthesis models

Torchgeo ⭐ 1,788

TorchGeo: datasets, samplers, transforms, and pre-trained models for geospatial data

Pytorch Retinanet ⭐ 1,764

Pytorch implementation of RetinaNet object detection.

Wavernn ⭐ 1,761

WaveRNN Vocoder + TTS

3d Pointcloud ⭐ 1,753

Papers and Datasets about Point Cloud.

Impersonator ⭐ 1,630

PyTorch implementation of our ICCV 2019 paper: Liquid Warping GAN: A Unified Framework for Human Motion Imitation, Appearance Transfer and Novel View Synthesis

Petastorm ⭐ 1,614

Petastorm library enables single machine or distributed training and evaluation of deep learning models from datasets in Apache Parquet format. It supports ML frameworks such as Tensorflow, Pytorch, and PySpark and can be used from pure Python code.

Pytorch Cpp ⭐ 1,559

C++ Implementation of PyTorch Tutorials for Everyone

Danet ⭐ 1,556

Dual Attention Network for Scene Segmentation (CVPR2019)

Transfer Learning Conv Ai ⭐ 1,499

🦄 State-of-the-Art Conversational AI with Transfer Learning

Senet.pytorch ⭐ 1,494

PyTorch implementation of SENet

Cogdl ⭐ 1,437

CogDL: A Comprehensive Library for Graph Deep Learning (WWW 2023)

Efficientdet.pytorch ⭐ 1,408

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

Open3d Ml ⭐ 1,389

An extension of Open3D to address 3D Machine Learning tasks

Insightface_pytorch ⭐ 1,336

Pytorch0.4.1 codes for InsightFace

Tutorials ⭐ 1,332

MONAI Tutorials

Dg Net ⭐ 1,235

👫 Joint Discriminative and Generative Learning for Person Re-identification. CVPR'19 (Oral) 👫

Animegan ⭐ 1,225

A simple PyTorch Implementation of Generative Adversarial Networks, focusing on anime face drawing.

Fastmri ⭐ 1,148

A large-scale dataset of both raw MRI measurements and clinical MRI images.

Yolact_edge ⭐ 1,138

The first competitive instance segmentation approach that runs on small edge devices at real-time speeds.

Bisenet ⭐ 1,130

Add bisenetv2. My implementation of BiSeNet

A Heterogeneous Benchmark for Information Retrieval. Easy to use, evaluate your models across 15+ diverse IR datasets.

Gan Compression ⭐ 1,028

[CVPR 2020] GAN Compression: Efficient Architectures for Interactive Conditional GANs

Transformer ⭐ 1,014

PyTorch Implementation of "Attention Is All You Need"

Textbox ⭐ 966

TextBox 2.0 is a text generation library with pre-trained language models

Cdial Gpt ⭐ 944

A Large-scale Chinese Short-Text Conversation Dataset and Chinese pre-training dialog models

Discogan Pytorch ⭐ 928

PyTorch implementation of "Learning to Discover Cross-Domain Relations with Generative Adversarial Networks"

Torchmoji ⭐ 882

😇A pyTorch implementation of the DeepMoji model: state-of-the-art deep learning model for analyzing sentiment, emotion, sarcasm etc

Medmnist ⭐ 818

[pip install medmnist] 18 MNIST-like Datasets for 2D and 3D Biomedical Image Classification

Tfrecord ⭐ 811

TFRecord reader for PyTorch

Pytorch Video Recognition ⭐ 798

PyTorch implemented C3D, R3D, R2Plus1D models for video activity recognition.

Efficient Pytorch ⭐ 769

My best practice of training large dataset using PyTorch.

Ganomaly ⭐ 767

GANomaly: Semi-Supervised Anomaly Detection via Adversarial Training

Torchxrayvision ⭐ 708

TorchXRayVision: A library of chest X-ray datasets and models. Classifiers, segmentation, and autoencoders.

Deepsvg ⭐ 705

[NeurIPS 2020] Official code for the paper "DeepSVG: A Hierarchical Generative Network for Vector Graphics Animation". Includes a PyTorch library for deep learning with SVG data.

Transformer ⭐ 691

Implementation of Transformer model (originally from Attention is All You Need) applied to Time Series.

Self Driving Car In Video Games ⭐ 687

A deep neural network that learns to drive in video games

Pose Transfer ⭐ 672

Code for the paper Progressive Pose Attention for Person Image Generation in CVPR19 (Oral).

Lightgcn Pytorch ⭐ 642

The PyTorch implementation of LightGCN

Voicefilter ⭐ 640

Unofficial PyTorch implementation of Google AI's VoiceFilter system

Facial Similarity With Siamese Networks In Pytorch ⭐ 620

Implementing Siamese networks with a contrastive loss for similarity learning

Pytorch Segmentation Toolbox ⭐ 616

PyTorch Implementations for DeeplabV3 and PSPNet

3dmppe_posenet_release ⭐ 616

Official PyTorch implementation of "Camera Distance-aware Top-down Approach for 3D Multi-person Pose Estimation from a Single RGB Image", ICCV 2019

Timecycle ⭐ 612

Learning Correspondence from the Cycle-consistency of Time (CVPR 2019)

Nas Bench 201 ⭐ 610

NAS-Bench-201 API and Instruction

Fots.pytorch ⭐ 604

FOTS Pytorch Implementation

D2-Net: A Trainable CNN for Joint Description and Detection of Local Features

Streaming ⭐ 597

A Data Streaming Library for Efficient Neural Network Training

Dexined ⭐ 570

DexiNed: Dense EXtreme Inception Network for Edge Detection

Dataloaders_dali ⭐ 569

PyTorch DataLoaders implemented with DALI for accelerating image preprocessing

Alignedreid Re Production Pytorch ⭐ 554

Reproduce AlignedReID: Surpassing Human-Level Performance in Person Re-Identification, using Pytorch.

Brain Segmentation Pytorch ⭐ 554

U-Net implementation in PyTorch for FLAIR abnormality segmentation in brain MRI

Liteflownet ⭐ 543

LiteFlowNet: A Lightweight Convolutional Neural Network for Optical Flow Estimation, CVPR 2018 (Spotlight paper, 6.6%)

Pytorch Medical Segmentation ⭐ 536

This repository is an unoffical PyTorch implementation of Medical segmentation in 2D and 3D.

Howtotrainyourmamlpytorch ⭐ 530

The original code for the paper "How to train your MAML" along with a replication of the original "Model Agnostic Meta Learning" (MAML) paper in Pytorch.

Mmdetection Annotated ⭐ 522

mmdetection源码注释

Pytorch Custom Dataset Examples ⭐ 518

Some custom dataset examples for PyTorch

Deeplabv3plus Pytorch ⭐ 511

Here is a pytorch implementation of deeplabv3+ supporting ResNet(79.155%) and Xception(79.945%). Multi-scale & flip test and COCO dataset interface has been finished.

Daisyrec ⭐ 496

This is the repository of our article published in RecSys 2020 "Are We Evaluating Rigorously? Benchmarking Recommendation for Reproducible Evaluation and Fair Comparison" and of several follow-up studies.

Pytorch Progressive_growing_of_gans ⭐ 485

PyTorch implementation of Progressive Growing of GANs for Improved Quality, Stability, and Variation.

Attention Networks For Classification ⭐ 477

Hierarchical Attention Networks for Document Classification in PyTorch

Pytorch Image Classification ⭐ 475

Tutorials on how to implement a few key architectures for image classification using PyTorch and TorchVision.

Complete Life Cycle Of A Data Science Project ⭐ 472

Complete-Life-Cycle-of-a-Data-Science-Project

"Learning Delicate Local Representations for Multi-Person Pose Estimation" (ECCV 2020 Spotlight)

Pytorch Flows ⭐ 444

PyTorch implementations of algorithms for density estimation

[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.

This is a pytorch re-implementation of EAST: An Efficient and Accurate Scene Text Detector.

Laneatt ⭐ 418

Code for the paper entitled "Keep your Eyes on the Lane: Real-time Attention-guided Lane Detection" (CVPR 2021)

Gaitset ⭐ 411

A flexible, effective and fast cross-view gait recognition network

Torch Toolbox ⭐ 400

🛠 Toolbox to extend PyTorch functionalities

Related Searches

Python Dataset (15,256)

Python Pytorch (14,667)

Deep Pytorch (7,270)

Jupyter Notebook Dataset (6,824)

Jupyter Pytorch (4,791)

Deep Learning Pytorch (4,652)

Jupyter Notebook Pytorch (3,877)

Machine Learning Pytorch (2,776)

Deep Learning Dataset (2,499)

Deep Dataset (2,364)

1-100 of 1,516 search results

Privacy | About | Terms | Follow Us On Twitter

Copyright 2018-2023 Awesome Open Source. All rights reserved.