Cómo dibujar la estructura de red del archivo prototxt

Pip instala pydotplus (es necesario instalar el paquete pydotplus)

En el bash del contenedor, usamos:

python caffe/python/draw _ net py workspace/pretty net proto txt workspace. /pretty net .jpg 11

espacio de trabajo/pretty net .

Workspace/prettynet.jpg: el nombre (y ubicación) del ipg de destino.

O código.

#Cargar módulo

Café importado

Sorteo de café importado

Importar caffe_pb2 desde caffe.proto

Importar formato de texto desde google.protobuf

#Establecer parámetros

Archivo de protocolo de red de entrada = 'Archivo de protocolo de red de prueba' #Archivo de configuración de red

salida _ imagen _ file = 'test_net jpg' #Imagen de salida

rankdir ='LR' # Método de disposición de la red: LR, TB, RL

#Red de lectura

net=caffe_pb2. NetworkParameters()

Text_format. merge(open(input _ net _ proto _ file).read(), net)

#Cerrar la red

Imprimir ('Dibujar la red en s' output_image_file ')

caffe.draw.draw_net_to_file(net, salida_image_file, rankingdir)

Imprimir('Completo...')