All Categories → No Category → self-supervised-learning

Top 83 self-supervised-learning open source projects

SimCLRv2 - Big Self-Supervised Models are Strong Semi-Supervised Learners

✭ 2,720

Jupyter Notebook python computer-vision unsupervised-learning representation-learning self-supervised-learning simclr contrastive-learning simclrv2

Sfmlearner

An unsupervised learning framework for depth and ego-motion estimation from monocular videos

✭ 1,661

Jupyter Notebook python deep-learning unsupervised-learning visual-odometry depth-prediction self-supervised-learning

Transferlearning

Transfer learning / domain adaptation / domain generalization / multi-task learning etc. Papers, codes, datasets, applications, tutorials.-迁移学习

Pytorch Metric Learning

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

✭ 3,936

python deep-learning machine-learning pytorch computer-vision image-retrieval metric-learning deep-metric-learning self-supervised-learning contrastive-learning

adareg-monodispnet

Repository for Bilateral Cyclic Constraint and Adaptive Regularization for Unsupervised Monocular Depth Prediction (CVPR2019)

✭ 22

python computer-vision adaptive-learning unsupervised-learning cvpr 3d-reconstruction 3d-vision depth-prediction self-supervised-learning single-image-depth-prediction

learning-topology-synthetic-data

Tensorflow implementation of Learning Topology from Synthetic Data for Unsupervised Depth Completion (RAL 2021 & ICRA 2021)

✭ 22

python shell machine-learning computer-vision deep-learning tensorflow void depth unsupervised-learning sensor-fusion ucla 3d-reconstruction ral depth-estimation 3d-vision kitti icra self-supervised-learning depth-completion

SIGIR2021 Conure

One Person, One Model, One World: Learning Continual User Representation without Forgetting

TWIST

Official codes: Self-Supervised Learning by Estimating Twin Class Distribution

✭ 72

python computer-vision deep-learning twist self-supervised-learning pretraining

SHOT-plus

code for our TPAMI 2021 paper "Source Data-absent Unsupervised Domain Adaptation through Hypothesis Transfer and Labeling Transfer"

✭ 46

python shell semi-supervised-learning transfer-learning domain-adaptation self-supervised-learning multi-source-domain-adaptation semi-supervised-domain-adaptation tpami black-box-method partial-set-domain-adaptation

radar-ml

Detect (classify and localize) people, pets and objects using millimeter-wave radar.

✭ 47

python radar object-detection self-supervised-learning walabot-radar radar-datasets

PiCIE

PiCIE: Unsupervised Semantic Segmentation using Invariance and Equivariance in clustering (CVPR2021)

✭ 102

Jupyter Notebook python shell unsupervised-learning semantic-segmentation unsupervised-clustering self-supervised-learning

PIC

Parametric Instance Classification for Unsupervised Visual Feature Learning, NeurIPS 2020

✭ 41

python shell pytorch unsupervised-learning self-supervised-learning contrastive-learning

SSL CR Histo

Official code for "Self-Supervised driven Consistency Training for Annotation Efficient Histopathology Image Analysis" Published in Medical Image Analysis (MedIA) Journal, Oct, 2021.

✭ 32

python shell deep-learning semi-supervised-learning digital-pathology self-supervised-learning camelyon16 histopathology breastpathq teacher-student-training annotation-efficient

IFM

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

✭ 33

python deep-learning pytorch self-supervised-learning

self-supervised

Whitening for Self-Supervised Representation Learning | Official repository

✭ 83

python Dockerfile representation-learning self-supervised-learning

Co-mining

Co-mining: Self-Supervised Learning for Sparsely Annotated Object Detection, AAAI 2021.

✭ 16

python object-detection self-supervised-learning sparsely-annotated

al-fk-self-supervision

Official PyTorch code for CVPR 2020 paper "Deep Active Learning for Biased Datasets via Fisher Kernel Self-Supervision"

✭ 28

python image-classification image-recognition unsupervised-learning active-learning fisher-vectors self-supervised-learning

simclr-pytorch

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

✭ 89

Jupyter Notebook python deep-learning pytorch representation-learning self-supervised-learning pytorch-implementation contrastive-learning

video-clip-order-prediction

Self-supervised Spatiotemporal Learning via Video Clip Order Prediction

✭ 85

python video pytorch spatiotemporal self-supervised-learning cvpr2019

DIG

A library for graph deep learning research

✭ 1,078

python deep-learning graph-generation explainable-ml self-supervised-learning 3d-graph graph-neural-network

EgoNet

Official project website for the CVPR 2021 paper "Exploring intermediate representation for monocular vehicle pose estimation"

✭ 111

python C++deep-learning representation-learning vehicles autonomous-driving pose-estimation 3d-vision kitti projective-geometry 3d-pose-estimation self-supervised-learning 3d-object-detection orientation-estimation kitti-3d robot-perception cvpr2021

awesome-contrastive-self-supervised-learning

A comprehensive list of awesome contrastive self-supervised learning papers.

✭ 748

deep-learning transfer-learning representation-learning unsupervised-learning self-supervised-learning hacktoberfest-accepted

PASSL

PASSL包含 SimCLR，MoCo v1/v2，BYOL，CLIP，PixPro，BEiT，MAE等图像自监督算法以及 Vision Transformer，DEiT，Swin Transformer，CvT，T2T-ViT，MLP-Mixer，XCiT，ConvNeXt，PVTv2 等基础视觉算法

✭ 134

python deep-learning vit clip paddle pvt mae moco self-supervised-learning cvt simclr beit vision-transformer deit pixpro moco-v2 swin-transformer xcit convnext

CIKM2020-S3Rec

Code for CIKM2020 "S3-Rec: Self-Supervised Learning for Sequential Recommendation with Mutual Information Maximization"

✭ 150

python mutual-information self-supervised-learning sequential-recommendation

SelfTask-GNN

Implementation of paper "Self-supervised Learning on Graphs:Deep Insights and New Directions"

✭ 78

python shell machine-learning deep-learning graph-mining graph-neural-networks self-supervised-learning

GCA

[WWW 2021] Source code for "Graph Contrastive Learning with Adaptive Augmentation"

✭ 69

python deep-learning pytorch self-supervised-learning graph-representation-learning contrastive-learning graph-contrastive-learning

AdCo

AdCo: Adversarial Contrast for Efficient Learning of Unsupervised Representations from Self-Trained Negative Adversaries

✭ 148

python self-supervised-learning contrastive-learning adco

point-cloud-prediction

Self-supervised Point Cloud Prediction Using 3D Spatio-temporal Convolutional Networks

✭ 97

python cloud video point-cloud prediction point forecasting lidar range-image self-supervised-learning video-prediction point-cloud-forecasting point-cloud-prediction point-forecasting

SimCLR-in-TensorFlow-2

(Minimally) implements SimCLR (https://arxiv.org/abs/2002.05709) in TensorFlow 2.

✭ 75

Jupyter Notebook deep-learning tensorflow keras unsupervised-learning self-supervised-learning constrastive-learning

improving segmentation with selfsupervised depth

[CVPR21] Implementation of our work "Three Ways to Improve Semantic Segmentation with Self-Supervised Depth Estimation"

✭ 189

python semi-supervised-learning semantic-segmentation depth-estimation self-supervised-learning

sc depth pl

Pytorch Lightning Implementation of SC-Depth (V1, V2...) for Unsupervised Monocular Depth Estimation.

✭ 86

python shell computer-vision deep-learning pose-estimation depth-estimation self-supervised-learning

FisheyeDistanceNet

✭ 33

self-driving-car depth fisheye depth-estimation fisheye-lens-distortion depth-prediction monocular-depth self-supervised-learning monocular-depth-estimation fisheye-dataset fisheye-image icra2020

SelfSupervisedLearning-DSM

code for AAAI21 paper "Enhancing Unsupervised Video Representation Learning by Decoupling the Scene and the Motion“

✭ 26

python shell self-supervised-learning aaai2021

FKD

A Fast Knowledge Distillation Framework for Visual Recognition

✭ 49

python deep-learning pytorch knowledge-distillation efficient-algorithm distillation self-supervised-learning efficient-training efficientnet-pytorch

video repres mas

code for CVPR-2019 paper: Self-supervised Spatio-temporal Representation Learning for Videos by Predicting Motion and Appearance Statistics

✭ 63

python video tensorflow action-recognition spatio-temporal-analysis self-supervised-learning cvpr2019

MiniVox

Code for our ACML and INTERSPEECH papers: "Speaker Diarization as a Fully Online Bandit Learning Problem in MiniVox".

✭ 15

Cuda matlab C++python shell c M paper speaker-recognition online-learning speaker-diarization contextual-bandits bandit-algorithms interspeech self-supervised-learning acml interspeech2020

G-SimCLR

This is the code base for paper "G-SimCLR : Self-Supervised Contrastive Learning with Guided Projection via Pseudo Labelling" by Souradip Chakraborty, Aritra Roy Gosthipaty and Sayak Paul.

✭ 69

Jupyter Notebook clustering keras-tensorflow visual-recognition self-supervised-learning tensorflow2 contrastive-learning

Self-Supervised-Embedding-Fusion-Transformer

The code for our IEEE ACCESS (2020) paper Multimodal Emotion Recognition with Transformer-Based Self Supervised Feature Fusion.

✭ 57

python Cuda bert emotion-recognition multimodal-sentiment-analysis multimodal-deep-learning self-supervised-learning multimodal-emotion-recognition

BYOL

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

✭ 102

python deep-learning pytorch self-supervised-learning byol

CVPR21 PASS

PyTorch implementation of our CVPR2021 (oral) paper "Prototype Augmentation and Self-Supervision for Incremental Learning"

✭ 55

python incremental-learning lifelong-learning continual-learning self-supervised-learning

MSF

Official code for "Mean Shift for Self-Supervised Learning"

✭ 42

python shell machine-learning computer-vision convolutional-neural-networks representation-learning knn mean-shift self-supervised-learning

BossNAS

(ICCV 2021) BossNAS: Exploring Hybrid CNN-transformers with Block-wisely Self-supervised Neural Architecture Search

✭ 125

python shell transformer nas automl neural-architecture-search self-supervised-learning hybrid-neural-network

simsiam-cifar10

Code to train the SimSiam model on cifar10 using PyTorch

✭ 33

python machine-learning computer-vision deep-learning pytorch self-supervised-learning

newt

Natural World Tasks

✭ 24

python machine-learning computer-vision classification datasets self-supervised-learning

SoCo

[NeurIPS 2021 Spotlight] Aligning Pretraining for Detection via Object-Level Contrastive Learning

✭ 125

python shell Dockerfile object-detection self-supervised-learning contrastive-learning neurips-2021

GCL

List of Publications in Graph Contrastive Learning

✭ 25

graph self-supervised-learning gnn contrastive-learning

esvit

EsViT: Efficient self-supervised Vision Transformers

✭ 323

python shell self-supervised-learning vision-transformers

mae-scalable-vision-learners

A TensorFlow 2.x implementation of Masked Autoencoders Are Scalable Vision Learners

✭ 54

Jupyter Notebook keras autoencoder self-supervised-learning tensorflow2 masked-image-modeling

pillar-motion

Self-Supervised Pillar Motion Learning for Autonomous Driving (CVPR 2021)

✭ 98

python autonomous-driving lidar-point-cloud self-supervised-learning nuscenes point-cloud-motion

Awesome-Vision-Transformer-Collection

Variants of Vision Transformer and its downstream tasks

✭ 124

tracking awesome computer-vision deep-learning detection backbone generative-model segmentation pose-estimation lidar-point-cloud explainable-ai self-attention self-supervised-learning

awesome-graph-self-supervised-learning-based-recommendation

A curated list of awesome graph & self-supervised-learning-based recommendation.

✭ 37

machine-learning deep-learning recommendation-system graph-neural-networks self-supervised-learning knowledge-graph-for-recommendation contrastive-learning graph-based-recommendation

info-nce-pytorch

PyTorch implementation of the InfoNCE loss for self-supervised learning.

✭ 160

python contrastive-loss self-supervised-learning contrastive-learning

GeDML

Generalized Deep Metric Learning.

✭ 30

python computer-vision metric-learning loss-functions deep-metric-learning self-supervised-learning contrastive-learning

mmselfsup

OpenMMLab Self-Supervised Learning Toolbox and Benchmark

✭ 2,315

python shell Dockerfile pytorch unsupervised-learning moco self-supervised-learning simclr deepcluster simsiam swav

TCE

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

✭ 51

python computer-vision deep-learning pytorch embeddings metric-learning representation-learning action-recognition hmdb51 contrastive-loss self-supervised-learning ucf-101 kinetics-datasets contrastive-learning tsne-visualisations

self6dpp

Self6D++: Occlusion-Aware Self-Supervised Monocular 6D Object Pose Estimation. IEEE Transactions on Pattern Analysis and Machine Intelligence (TPAMI) 2021.

✭ 45