Awesome Open Source

Programming Languages

Search results for pytorch dataset

911 search results found

Pytorch Cyclegan And Pix2pix ⭐ 21,090

Image-to-Image Translation in PyTorch

Insightface ⭐ 19,487

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

Vision ⭐ 15,059

Datasets, Transforms and Models specific to Computer Vision

Cvat ⭐ 13,864

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.

Techniques ⭐ 9,411

Techniques for deep learning with satellite & aerial imagery

Deeplake ⭐ 8,650

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

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

Yet Another Efficientdet Pytorch ⭐ 4,892

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

Yolact ⭐ 4,793

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

Nerf Pytorch ⭐ 4,711

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

Nnunet ⭐ 4,608

Mmdetection3d ⭐ 4,497

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

Deep Person Reid ⭐ 4,029

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,134

LPIPS metric. pip install lpips

Recbole ⭐ 3,038

A unified, comprehensive and efficient recommendation library

Super Slomo ⭐ 2,873

PyTorch implementation of Super SloMo by Jiang et al.

[CVPR2020] Adversarial Latent Autoencoders

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.

Torchgeo ⭐ 2,067

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

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,906

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

Pytorch Retinanet ⭐ 1,764

Pytorch implementation of RetinaNet object detection.

Wavernn ⭐ 1,761

WaveRNN Vocoder + TTS

Pytorch Cpp ⭐ 1,710

C++ Implementation of PyTorch Tutorials for Everyone

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

Danet ⭐ 1,556

Dual Attention Network for Scene Segmentation (CVPR2019)

Open3d Ml ⭐ 1,530

An extension of Open3D to address 3D Machine Learning tasks

Transfer Learning Conv Ai ⭐ 1,499

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

Senet.pytorch ⭐ 1,494

PyTorch implementation of SENet

Tutorials ⭐ 1,489

MONAI Tutorials

Efficientdet.pytorch ⭐ 1,408

Implementation EfficientDet: Scalable and Efficient Object Detection in PyTorch

Insightface_pytorch ⭐ 1,336

Pytorch0.4.1 codes for InsightFace

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,197

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

Pytorch Video Recognition ⭐ 1,114

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

Gan Compression ⭐ 1,063

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

Transformer ⭐ 1,014

PyTorch Implementation of "Attention Is All You Need"

Data Juicer ⭐ 994

A one-stop data processing system to make data higher-quality, juicier, and more digestible for LLMs! 🍎 🍋 🌽 ➡️ ➡️🍸 🍹 🍷为大语言模型提供更高质量、更丰富、更易”消化“的数据！

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

Deepsvg ⭐ 829

[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.

Tfrecord ⭐ 811

TFRecord reader for PyTorch

Efficient Pytorch ⭐ 769

My best practice of training large dataset using PyTorch.

Ganomaly ⭐ 767

GANomaly: Semi-Supervised Anomaly Detection via Adversarial Training

Torchxrayvision ⭐ 760

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

Streaming ⭐ 755

A Data Streaming Library for Efficient Neural Network Training

Self Driving Car In Video Games ⭐ 729

A deep neural network that learns to drive in video games

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

Howtotrainyourmamlpytorch ⭐ 707

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.

Transformer ⭐ 691

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

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

Nas Bench 201 ⭐ 626

NAS-Bench-201 API and Instruction

Facial Similarity With Siamese Networks In Pytorch ⭐ 620

Implementing Siamese networks with a contrastive loss for similarity learning

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

Pytorch Segmentation Toolbox ⭐ 616

PyTorch Implementations for DeeplabV3 and PSPNet

Timecycle ⭐ 612

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

Fots.pytorch ⭐ 604

FOTS Pytorch Implementation

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

Pytorch Medical Segmentation ⭐ 536

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

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.

Complete Life Cycle Of A Data Science Project ⭐ 499

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

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

Ganilla ⭐ 476

Official Pytorch implementation of GANILLA

Pytorch Image Classification ⭐ 475

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

"Learning Delicate Local Representations for Multi-Person Pose Estimation" (ECCV 2020 Spotlight) & (COCO 2019 Keypoint Challenge Winner and Best Paper Award)

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

Openai Clip ⭐ 404

Simple implementation of OpenAI CLIP model in PyTorch.

Carefree Learn ⭐ 400

Deep Learning ❤️ PyTorch

Meta Sr Pytorch ⭐ 399

Meta-SR: A Magnification-Arbitrary Network for Super-Resolution (CVPR2019)

CRSLab is an open-source toolkit for building Conversational Recommender System (CRS).

Universenet ⭐ 392

USB: Universal-Scale Object Detection Benchmark (BMVC 2022)

Residualmaskingnetwork ⭐ 387

ICPR 2020: Facial Expression Recognition using Residual Masking Network

3dmppe_rootnet_release ⭐ 386

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

Paperrobot ⭐ 384

Code for PaperRobot: Incremental Draft Generation of Scientific Ideas

Pytorch Superpoint ⭐ 376

Superpoint Implemented in PyTorch: https://arxiv.org/abs/1712.07629

Pytorch Cyclegan ⭐ 371

A clean and readable Pytorch implementation of CycleGAN

Related Searches

Python Pytorch (15,943)

Python Dataset (14,962)

Deep Learning Pytorch (7,533)

Jupyter Notebook Dataset (6,824)

Jupyter Notebook Pytorch (4,892)

Machine Learning Pytorch (2,934)

Deep Learning Dataset (2,364)

Machine Learning Dataset (2,279)

Pytorch Convolutional Neural Networks (1,777)

Pytorch Neural Network (1,631)

1-100 of 911 search results

Privacy | About | Terms | Follow Us On Twitter

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