zoukankan      html  css  js  c++  java
  • Caffe---Pycaffe进行网络结构(xxx.prototxt)可视化

    Pycaffe---进行网络结构(xxx.prototxt)可视化

    解决网络结构(xxx.prototxt)可视化,还可以借助python接口,编写一个类似如下的pycaffe_draw_net.py:

    # -*- coding: utf-8 -*-
    #!/usr/bin/env python 
    
    import caffe  
    import caffe.draw  
    from caffe.proto import caffe_pb2  
    from google.protobuf import text_format  
          
    # Set the parameters  
    input_net_file = '/home/wp/caffe/myself/road/prototxt_files/train_val.prototxt' # net configuration file  
    output_image_file ='/home/wp/caffe/myself/road/Log/road_net.jpg' # the output image  
    rankdir ='LR' # net arrange method: LR、TB、RL  
          
    # Read the net  
    net = caffe_pb2.NetParameter()  
    text_format.Merge(open(input_net_file).read(), net)  
          
    # Draw the net  
    print('Drawing net to %s' % output_image_file)  
    caffe.draw.draw_net_to_file(net, output_image_file, rankdir)  
    print('done...') 
    

    # cd caffe

    # python  ./xx/xx/pycaffe_draw_net.py

  • 相关阅读:
    oracle_case when
    sql记录
    修改tomcat默认编码
    sql server_TED(透明数据加密)
    web请求过程
    freemarker 标签循环list 以及获取下标值
    鼠标操作
    窗口切换
    日历控件
    xpath定位
  • 原文地址:https://www.cnblogs.com/carle-09/p/9087101.html
Copyright © 2011-2022 走看看