Log In Sign Up
Models:
David Feaster
DavidFeaster/
DR-UNet
0
Downloads: 1
Diff of /drunet/predict_segment.py [000000] .. [2824d6]

Switch to unified view

a b/drunet/predict_segment.py 

  1
  
import segment





  
  

  2
  





  
  

  3
  
if __name__ == '__main__':





  
  

  4
  
    Seg = segment.Segmentation()





  
  

  5
  
    # start predict





  
  

  6
  
    input_dir = r''  # Fill in the image path





  
  

  7
  
    save_dir = r''  # fill in save path





  
  

  8
  
    Seg.predict_and_save(input_dir, save_dir)