Skip to content

ninatu/SimCLR

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

120 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

SimCLR for pre-training VGG19 weights

Fork of PyTorch SimCLR adapted for pre-training VGG19 weights for usage in Medical Out-of-Distribution Analysis Challenge (MOOD).

See original code and documentation in https://github.com/sthalles/SimCLR

See a description of our solution to the MOOD Challenge in https://github.com/ninatu/mood_challenge

Usage

  1. Install the anomaly detection framework
pip install git+https://github.com/ninatu/mood_challenge.git
  1. Perform "Data Preparation" step. See git@github.com:ninatu/mood_challenge.git
  2. Put correct paths in config/exp_1_mood.yaml and run
python run.py configs/exp_1_mood_tmp.yaml

About

PyTorch implementation of SimCLR: A Simple Framework for Contrastive Learning of Visual Representations

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Jupyter Notebook 95.5%
  • Python 4.5%