src/labelmap.py

""" Mapping of class IDs to lables. """

DR_LABELMAP = {
    0: 'No DR',
    1: 'Mild',
    2: 'Moderate',
    3: 'Severe',
    4: 'Proliferative DR',
}