Skip to content

The Mask RCNN's application in IDRiD(only for challenge 1)

Notifications You must be signed in to change notification settings

yejg2017/IDRiD_Challenge_Mask-RCNN

Repository files navigation

IDRiD_Challenge_Mask-RCNN

the the data can be download here

The Mask RCNN's application in IDRiD(only for challenge 1)

原始的数据结构

  • Original:

    --traiing

     ---IDRiD_01.jpg
     
     ---IDRiD_02.jpg
     ...
    

    --testing ---IDRiD_01.jpg

     ---IDRiD_02.jpg
     ...
    
  • Groundtruths:

    --training

      ---Haemorrhages
        
        IDRiD_01.jpg
        
        IDRiD_02.jpg
        ...
      ---HardExudates
      
      ---Microaneurysms
      
      ---OpticDisc
      
      ---SoftExudates
      
    --testing
      ---Haemorrhages
        
        IDRiD_01.jpg
        
        IDRiD_02.jpg
        ...
      ---HardExudates
      
      ---Microaneurysms
      
      ---OpticDisc
      
      ---SoftExudates
    
  • Step 1 : Create Dataset

First,create directory data

mkdir data

And create data

python dataset.py

The data will store in data,they are too big,so I do not show that.

  • Step 2: Train the Model
python main.py

the checkpoint will be stored in directory checkpoint,you can download the pretrained model in here

  • Step 3: Demo Show the Mask and model evaluation

Diabetes

About

The Mask RCNN's application in IDRiD(only for challenge 1)

Topics

Resources

Stars

Watchers

Forks

Packages

No packages published