Awesome Open Source

Programming Languages

Search results for pytorch self supervised learning

self-supervised-learning x

157 search results found

Pytorch Metric Learning ⭐ 5,734

The easiest way to use deep metric learning in your application. Modular, flexible, and extensible. Written in PyTorch.

Mmpretrain ⭐ 3,114

OpenMMLab Pre-training Toolbox and Benchmark

Mmselfsup ⭐ 3,066

OpenMMLab Self-Supervised Learning Toolbox and Benchmark

Lightly ⭐ 2,665

A python library for self-supervised learning on images.

Byol Pytorch ⭐ 1,630

Usable Implementation of "Bootstrap Your Own Latent" self-supervised learning, from Deepmind, in Pytorch

Easycv ⭐ 1,614

An all-in-one toolkit for computer vision

Spark ⭐ 1,355

[ICLR'23 Spotlight🔥] The first successful BERT/MAE-style pretraining on any convolutional network; Pytorch impl. of "Designing BERT for Convolutional Networks: Sparse and Hierarchical Masked Modeling"

Solo Learn ⭐ 1,318

solo-learn: a library of self-supervised methods for visual representation learning powered by Pytorch Lightning

Torchssl ⭐ 1,260

A PyTorch-based library for semi-supervised learning (NeurIPS'21)

Training_extensions ⭐ 1,119

Train, Evaluate, Optimize, Deploy Computer Vision Models via OpenVINO™

Vitpose ⭐ 950

The official repo for [NeurIPS'22] "ViTPose: Simple Vision Transformer Baselines for Human Pose Estimation" and [Arxiv'22] "ViTPose+: Vision Transformer Foundation Model for Generic Body Pose Estimation"

Videomae ⭐ 864

[NeurIPS 2022 Spotlight] VideoMAE: Masked Autoencoders are Data-Efficient Learners for Self-Supervised Video Pre-Training

Vl Bert ⭐ 680

Code for ICLR 2020 paper "VL-BERT: Pre-training of Generic Visual-Linguistic Representations".

All-in-one Toolbox for Computer Vision Research.

Openmixup ⭐ 538

CAIRI Supervised, Semi- and Self-Supervised Visual Representation Learning Toolbox and Benchmark

Lemniscate.pytorch ⭐ 467

Unsupervised Feature Learning via Non-parametric Instance Discrimination

Pytorch Dense Correspondence ⭐ 460

Code for "Dense Object Nets: Learning Dense Visual Object Descriptors By and For Robotic Manipulation"

Openstl ⭐ 450

OpenSTL: A Comprehensive Benchmark of Spatio-Temporal Predictive Learning

Manydepth ⭐ 440

[CVPR 2021] Self-supervised depth estimation from short sequences

Self Supervised Depth Completion ⭐ 418

ICRA 2019 "Self-supervised Sparse-to-Dense: Self-supervised Depth Completion from LiDAR and Monocular Camera"

Epipolarpose ⭐ 366

Self-Supervised Learning of 3D Human Pose using Multi-view Geometry (CVPR2019)

Declutr ⭐ 349

The corresponding code from our paper "DeCLUTR: Deep Contrastive Learning for Unsupervised Textual Representations". Do not hesitate to open an issue if you run into any trouble!

Hierarchical Image Pyramid Transformer - CVPR 2022 (Oral)

Voice Conversion With Just Nearest Neighbors

Scenerf ⭐ 284

[ICCV 2023] Official implementation of "SceneRF: Self-Supervised Monocular 3D Scene Reconstruction with Radiance Fields"

Problem Agnostic Speech Encoder

LibAUC: A Deep Learning Library for X-Risk Optimization

Graphmae ⭐ 253

GraphMAE: Self-Supervised Masked Graph Autoencoders in KDD'22

Dsmil Wsi ⭐ 224

DSMIL: Dual-stream multiple instance learning networks for tumor detection in Whole Slide Image

Iseebetter ⭐ 223

iSeeBetter: Spatio-Temporal Video Super Resolution using Recurrent-Generative Back-Projection Networks | Python3 | PyTorch | GANs | CNNs | ResNets | RNNs | Published in Springer Journal of Computational Visual Media, September 2020, Tsinghua University Press

Contrastivecrop ⭐ 202

[CVPR 2022 Oral] Crafting Better Contrastive Views for Siamese Representation Learning

Tadaconv ⭐ 177

[ICLR 2022] TAda! Temporally-Adaptive Convolutions for Video Understanding. This codebase provides solutions for video classification, video representation learning and temporal detection.

Official code for our ECCV'22 paper "A Fast Knowledge Distillation Framework for Visual Recognition"

Code for the paper "Adversarial Self-supervised Contrastive Learning" (NeurIPS 2020)

Latent Pose Reenactment ⭐ 145

The authors' implementation of the "Neural Head Reenactment with Latent Pose Descriptors" (CVPR 2020) paper.

Simclr Pytorch ⭐ 140

PyTorch implementation of SimCLR: supports multi-GPU training and closely reproduces results

Lamda Ssl ⭐ 130

30 Semi-Supervised Learning Algorithms

Greenmim ⭐ 129

[NeurIPS2022] Official implementation of the paper 'Green Hierarchical Vision Transformer for Masked Image Modeling'.

Data2vec Pytorch ⭐ 126

PyTorch implementation of "data2vec: A General Framework for Self-supervised Learning in Speech, Vision and Language" from Meta AI

[AAAI 2022] An official source code for paper Deep Graph Clustering via Dual Correlation Reduction.

This is the official repository for ICRA-2023 paper "STEPS: Joint Self-supervised Nighttime Image Enhancement and Depth Estimation"

[CVPR 2020] Action Segmentation with Joint Self-Supervised Temporal Domain Adaptation (PyTorch)

Un-Mix: Rethinking Image Mixtures for Unsupervised Visual Representation Learning.

Co Learning Learning From Noisy Labels With Self Supervision ⭐ 107

The official implementation of the ACM MM'2021 paper Co-learning: Learning from noisy labels with self-supervision.

Iccv_maet ⭐ 104

[ICCV 2021] Multitask AET with Orthogonal Tangent Regularity for Dark Object Detection. A self-supervised learning way for low-light image object detection.

Self-Supervised Learning in PyTorch

Pixel Level Contrastive Learning ⭐ 95

Implementation of Pixel-level Contrastive Learning, proposed in the paper "Propagate Yourself", in Pytorch

The official PyTorch implementation of the paper "Learning by Analogy: Reliable Supervision from Transformations for Unsupervised Optical Flow Estimation".

Pytorch Byol ⭐ 94

PyTorch implementation of Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

Unofficial implementation with pytorch DistributedDataParallel for "MoCo: Momentum Contrast for Unsupervised Visual Representation Learning"

Autoregressive Predictive Coding ⭐ 87

Autoregressive Predictive Coding: An unsupervised autoregressive model for speech representation learning

Open source Python library for building bioimage analysis pipelines

Video Clip Order Prediction ⭐ 85

Self-supervised Spatiotemporal Learning via Video Clip Order Prediction

Torchflare ⭐ 83

TorchFlare is a simple, beginner-friendly, and easy-to-use PyTorch Framework train your models effortlessly.

Pytorch Lifestream ⭐ 79

A library built upon PyTorch for building embeddings on discrete event sequences using self-supervision

Self Supervised Relational Reasoning ⭐ 78

Official PyTorch implementation of the paper "Self-Supervised Relational Reasoning for Representation Learning", NeurIPS 2020 Spotlight.

Lossyless ⭐ 76

Generic image compressor for machine learning. Pytorch code for our paper "Lossy compression for lossless prediction".

Self-supervised Product Quantization for Deep Unsupervised Image Retrieval - ICCV2021

Ssl_speech_restoration ⭐ 75

SelfRemaster: SSL Speech Restoration

Calibrated Backprojection Network ⭐ 74

PyTorch Implementation of Unsupervised Depth Completion with Calibrated Backprojection Layers (ORAL, ICCV 2021)

Monodepth_benchmark ⭐ 74

Code for "Deconstructing Monocular Depth Reconstruction: The Design Decisions that Matter" (https://arxiv.org/abs/2208.01489)

Implementation of "Molecular Contrastive Learning of Representations via Graph Neural Networks" in PyTorch Geometric

Semimtr Text Recognition ⭐ 68

Multimodal Semi-Supervised Learning for Text Recognition (SemiMTR)

Loss Gated Learning ⭐ 65

ICASSP 2022: 'Self-supervised Speaker Recognition with Loss-gated Learning'

Ssql Eccv2022 ⭐ 64

PyTorch implementation of SSQL (Accepted to ECCV2022 oral presentation)

Tri Depth ⭐ 61

[WACV 2023] Self-Supervised Monocular Depth Estimation: Solving the Edge-Fattening Problem

Cvf Sid_pytorch ⭐ 58

Official implementation of the paper "CVF-SID: Cyclic multi-Variate Function for Self-Supervised Image Denoising by Disentangling Noise from Image" (CVPR 2022)

Cytoself ⭐ 58

Self-supervised models for encoding protein localization patterns from microscopy images

Fusiondepth ⭐ 57

Official implementation for paper "Advancing Self-supervised Monocular Depth Learning with Sparse LiDAR"

Form2fit ⭐ 57

Train generalizable policies for kit assembly with self-supervised dense correspondence learning.

Self Supervised Vit Path ⭐ 50

Self-Supervised Vision Transformers Learn Visual Concepts in Histopathology (LMRL Workshop, NeurIPS 2021)

Audiossl ⭐ 50

A library built for easier audio self-supervised training, downstream tasks evaluation

Pytorch Scarf ⭐ 47

Implementation of SCARF: Self-Supervised Contrastive Learning using Random Feature Corruption in Pytorch, a model learning a representation of tabular data using contrastive learning. It is inspired from SimCLR and uses a similar architecture and loss.

Policy Adaptation During Deployment ⭐ 46

Training code and evaluation benchmarks for the "Self-Supervised Policy Adaptation during Deployment" paper.

Ssl Wearables ⭐ 45

Self-supervised learning for wearables using the UK-Biobank (>700,000 person-days)

Mixco Mixup Contrast ⭐ 45

MixCo: Mix-up Contrastive Learning for Visual Representation, NeurIPS 2020 Workshop on SSL

Lighthubert ⭐ 42

LightHuBERT: Lightweight and Configurable Speech Representation Learning with Once-for-All Hidden-Unit BERT

Dualrefine ⭐ 42

This repository contains the code for the paper "DualRefine: Self-Supervised Depth and Pose Estimation Through Iterative Epipolar Sampling and Refinement Toward Equilibrium", a self-supervised depth and pose estimation model. It refines depth and pose estimates by computing matching costs based on epipolar geometry and deep equilibrium model.

Point2vec ⭐ 40

Self-Supervised Representation Learning on Point Clouds (GCPR 2023 | T4V Workshop @ CVPR 2023)

This repository contains the code implementation used in the paper Temporally Coherent Embeddings for Self-Supervised Video Representation Learning (TCE).

Contrastive Learner ⭐ 39

A simple to use pytorch wrapper for contrastive self-supervised learning on any neural network

Bootstrap Your Own Latent: A New Approach to Self-Supervised Learning

Official PyTorch implementation of "PT4AL: Using Self-Supervised Pretext Tasks for Active Learning (ECCV2022)"

Cutpaste ⭐ 36

Unofficial implementation of Google "CutPaste: Self-Supervised Learning for Anomaly Detection and Localization" in PyTorch

An Empirical Investigation of Representation Learning for Imitation (EIRLI), NeurIPS'21

Code for paper "Can contrastive learning avoid shortcut solutions?" NeurIPS 2021.

Positional_cl ⭐ 33

code for paper Positional Contrastive Learning for Volumetric Medical Image Segmentation

Benchmarktransferlearning ⭐ 30

Official PyTorch Implementation and Pre-trained Models for Benchmarking Transfer Learning for Medical Image Analysis

[NeurIPS 2021] ORL: Unsupervised Object-Level Representation Learning from Scene Images

Parametric Instance Classification for Unsupervised Visual Feature Learning, NeurIPS 2020

Ssm Pytorch ⭐ 27

Towards Human-Machine Cooperation: Self-supervised Sample Mining for Object Detection

Decoder Denoising ⭐ 27

Pytorch reimplementation of Decoder Denoising Pretraining for Semantic Segmentation

Self Supervised Learning For Fine Grained Image Classification ⭐ 24

Applied Self Supervised Learning techniques such as Jigsaw as pretext task, SRGAN and SimCLR for fine-grained classification

Ne2ne Image Denoising ⭐ 23

Deep Unsupervised Image Denoising, based on Neighbour2Neighbour training

Understanding Cdfsl ⭐ 23

(NeurIPS 2022) Understanding Cross-Domain Few-Shot Learning Based on Domain Similarity and Few-Shot Difficulty

Emotion_detection_cpc ⭐ 22

Emotion detection in audio utilising self-supervised representations trained with Contrastive Predictive Coding (CPC).

Only Noisy Training ⭐ 22

A self-supervised speech denoising strategy named Only-Noisy Training (ONT), which solves the speech denoising problem with only noisy audio signals in audio space for the first time.

Freq Aware Depth ⭐ 21

[WACV2023] Frequency Aware Self-supervised Depth Estimation

Ssr_bmvc2022 ⭐ 21

SSR: An Efficient and Robust Framework for Learning with Unknown Label Noise (BMVC2022)

This repo contains the code for the CVPR 2023 paper: "CrOC : Cross-View Online Clustering for Dense Visual Representation Learning".

Related Searches

Python Pytorch (14,671)

Deep Learning Pytorch (7,533)

Jupyter Notebook Pytorch (4,892)

Machine Learning Pytorch (2,934)

Dataset Pytorch (1,847)

Pytorch Neural Network (1,631)

Pytorch Tensorflow (1,442)

Network Pytorch (1,299)

Pytorch Neural (1,217)

Pytorch Computer Vision (1,092)

1-100 of 157 search results

Privacy | About | Terms | Follow Us On Twitter

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