Skip to content

Latest commit

 

History

History
5 lines (4 loc) · 954 Bytes

File metadata and controls

5 lines (4 loc) · 954 Bytes

BIRA-NET-Bilinear-Attention-Net-for-Diabetic-Retinopathy-Grading

This is a PyTorch implementation of BIRA-NET paper. Link to the paper: https://arxiv.org/pdf/1905.06312.pdf You can find it's official implementation also but available repository does not have whole code. They have given only RA part of the model. There is one discrpancy in my code. I could not match the exact model structure after the part of outer product which is given in the last paragraph of section 3.3. There are two code files. One model has my modified CNN backbone to extract features (it contains dilated convolutions, you can edit it according to your use case). Other model has Resnet-18 as backbone.

biranet