Awesome Open Source
Search
Programming Languages
Languages
All Categories
Categories
About
Search results for pytorch dataset
dataset
x
pytorch
x
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
Tts
⭐
8,144
🤖 💬 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.
Text
⭐
3,411
Models, data loaders and abstractions for language processing, powered by PyTorch
Clue
⭐
3,345
中文语言理解测评基准 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.
Alae
⭐
2,850
[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
Beir
⭐
1,676
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
⭐
720
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.
Rsn
⭐
462
"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
Mirnet
⭐
436
[ECCV 2020] Learning Enriched Features for Real Image Restoration and Enhancement. SOTA results for image denoising, super-resolution, and image enhancement.
East
⭐
419
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
⭐
397
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
Next >
Privacy
|
About
|
Terms
|
Follow Us On Twitter
Copyright 2018-2025 Awesome Open Source. All rights reserved.