Installation (only used with Ubuntu-22.04) git clone https://github.com/kamilisjon/self-supervised.git cd self-supervised python3 -m venv venv source venv/bin/activate python3 -m pip install -e . Usage Prepare data Download data from https://www.kaggle.com/datasets/volodymyrgavrysh/pneumothorax-binary-classification-task Extract the download zip file Generate pre-text data python3 examples/data/pneumothorax.py Vizualize examples python3 examples/data/viz_classification_data.py Train PYTHONPATH=. python examples/train_effnet.py