Active Learning for Single Class Malignant Tumor Detection in Mammograms

Dataset Preparation

• Obtain the AIIMS data from sumanyu in a particular format.
• Make a data directory in the root of this project.
• Put the AIIMS data in this data directory.

Scripts Description

• aiims_eval.py: For finding fROC on AIIMS test data.
• train.py: For training on some percentage of training data.
• AL.py: For Active Learning iterations.

Dependencies

• Anaconda3
• Python>=3.6

Environment

• Create a conda env from .yml specified.

conda create env -f env.yml