神经网络边缘检测hed

整体嵌套的边缘检测：

Holistically-Nested Edge Detection

vgg16：

https://github.com/harsimrat-eyeem/holy-edge

c++版的：

https://github.com/harsimrat-eyeem/hed

需要建一个环境变量：OMP_NUM_THREADS

os.environ.get('OMP_NUM_THREADS')
 

也可以不用建环境变量，可以直接改为4

建默认参数：

  

   

    

     

    
    

     

      if __name__ == '__main__':
     
    
   

    

     

    
    

     
 
     
    
   

    

     

    
    

     

      # args.run_test=1
     
    
   

    

     

    
    

     

      # args.config_file="hed/configs/hed.yaml"
     
    
   

    

     

    
    

     

      # args.gpu_limit=0.4
     
    
   

    

     

    
    

     

      # args.download_data=1
     
    
   

    

     

    
    

     

      parser = argparse.ArgumentParser(description='Utility for Training/Testing DL models(Concepts/Captions) using theano/keras')
     
    
   

    

     

    
    

     

      parser.add_argument('--config-f
     
    
  
 

文章来源: blog.csdn.net，作者：网奇，版权归原作者所有，如需转载，请联系作者。

原文链接：blog.csdn.net/jacke121/article/details/90899358