Spaces:

kaust-generative-ai
/

diabetic-retinopathy

Running

App Files Files Community

diabetic-retinopathy / labelmap.py

DmitriiKhizbullin's picture

DmitriiKhizbullin

Initial implementation of trainer and gradio app

699342a 9 months ago

112 Bytes

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