This repository contains the Keras implementation using Tensorflow as backend of our paper "Progressive Multi-scale Consistent Network for Multi-class Fundus Lesion Segmentation"
python 3.6
numpy 1.16.4
keras 2.3.1
tensorflow 1.13.1
pillow 7.0.0
opencv-python 4.1.0
-
Clone the repository, and download the pre-trained ImaenNet model, put them into ./ folder. The details of the training are in train.py file.
-
And then run the code:python train.py Note that the parameters and paths should be set beforehand
-
Once the training is complete, you can run the test.py to test your model. Run the code : python test.py.
Code can only be used for ACADEMIC PURPOSES. NO COMERCIAL USE is allowed. Copyright © College of Computer Science, Nankai University. All rights reserved.