pablomorales92 / cnn_plus_vgpmil

This repository holds the code to train VGPMIL on CNN features

cnn_plus_vpmil

This repository holds the code to train VGPMIL on CNN features We use CNN features as the input for VGPMIL.

How to use

• To install all requirements please use anaconda/miniconda and type the following commands:
  • conda env create -f environment.yaml
  • conda activate cnn_plus_vgpmil
• Run the program in the root folder 'cnn_plus_vgpmil':
  • python src/main.py
• To change parameters, please use the 'config.yaml'