Optic-Net Classifier for Retinal Diseases
The authors propose a novel convolution neural network architecture to distinguish between different degeneration of retinal layers and their underlying causes. The proposed architecture includes a new residual unit subsuming atrous separable convolution, a novel building block to prevent gradient degradation in the residual unit where the middle (3×3) convolution operations are replaced with two different operations running in parallel: a (2×2) atrous convolution in the left branch and a (2×2) atrous separable convolution in the right branch. The proposed novel architecture is trained from scratch and outperforms other classification models while addressing the issue of gradient explosion, approaching near perfect accuracy of 99.8% and 100% for the Kermany and Srinivasan datasets, respectively.
Examples
Resource retrieval
Get the pre-trained net:
NetModel parameters
This model consists of a family of individual nets, each identified by a specific parameter combination. Inspect the available parameters:
Pick a non-default net by specifying the parameters:
Pick a non-default uninitialized net:
Basic usage
Classify a retinal image:
Obtain the probabilities predicted by the net:
Obtain the list of names of all available classes:
Net information
Obtain the total number of parameters:
Obtain the layer type counts:
Display the summary graphic:
Resource History
Reference
-
S. A. Kamran, S. Saha, A. S. Sabbir, A. Tavakkoli, "Optic-Net: A Novel Convolutional Neural Network for Diagnosis of Retinal Diseases from Optical Tomography Images," DOI: 10.1109/ICMLA.2019.00165 (2019)
- Available from: https://github.com/SharifAmit/OpticNet-71
-
Rights:
MIT License