Skip to content

Unsupervised learning of conv layers; feature extraction through image proximity

Notifications You must be signed in to change notification settings

Silent-Zebra/shallow-conv

Repository files navigation

Layer-Wise Contrastive Unsupervised Representation Learning

This repository contains code for the extended abstract in RUCS (link to be updated if/when published).

To learn the unsupervised filters, run Experiments_CIFAR100_1Layer.py.

To train a classifier from scratch (for supervised transfer), run Classifier_CIFAR100.py.

To then transfer and test learned filters, run Experiments_CIFAR10_FrozenConv.py.

Hyperparameters are within the respective files above; network structure hyperparameters are in networks.py.

About

Unsupervised learning of conv layers; feature extraction through image proximity

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages