def test_visualize(self): xgraph = XGraph() xgraph.add(XLayer( name='in1', type=['Input'], bottoms=[], tops=[], targets=[] )) xgraph.add(XLayer( name='in2', type=['Input'], bottoms=[], tops=[], targets=[] )) xgraph.add(XLayer( name='conv1', type=['Convolution'], bottoms=['in1'], tops=[], data=ConvData( weights=np.array([[[[1, 2], [3, 4]]]], dtype=np.float32), biases=np.array([0., 1.], dtype=np.float32) ), targets=[] )) xgraph.add(XLayer( name='add1', type=['Eltwise'], bottoms=['conv1', 'in2'], tops=[], targets=[] )) xgraph.insert(XLayer( name='conv2', type=['Convolution'], bottoms=['in2'], tops=['add1'], data=ConvData( weights=np.array([[[[1, 2], [3, 4]]]], dtype=np.float32), biases=np.array([0., 1.], dtype=np.float32) ), targets=[] )) xgraph.add(XLayer( name='conv3', type=['Convolution'], bottoms=['add1'], tops=[], data=ConvData( weights=np.array([[[[1, 2], [3, 4]]]], dtype=np.float32), biases=np.array([0., 1.], dtype=np.float32) ), targets=[] )) xgraph.add(XLayer( name='pool1', type=['Pooling'], bottoms=['add1'], tops=[], targets=[] )) xgraph.add(XLayer( name='add2', type=['Eltwise'], bottoms=['conv3', 'pool1'], tops=[], targets=[] )) assert len(xgraph) == 8 assert xgraph.get_layer_names() == \ ['in1', 'conv1', 'in2', 'conv2', 'add1', 'conv3', 'pool1', 'add2'] out_file = os.path.join(FILE_DIR, 'viz.png') xgraph.visualize(out_file) os.remove(out_file)
def test_visualize(self): xgraph = XGraph() xgraph.add( XLayer(name="in1", type=["Input"], bottoms=[], tops=[], targets=[])) xgraph.add( XLayer(name="in2", type=["Input"], bottoms=[], tops=[], targets=[])) xgraph.add( XLayer( name="conv1", type=["Convolution"], bottoms=["in1"], tops=[], data=ConvData( weights=np.array([[[[1, 2], [3, 4]]]], dtype=np.float32), biases=np.array([0.0, 1.0], dtype=np.float32), ), targets=[], )) xgraph.add( XLayer( name="add1", type=["Eltwise"], bottoms=["conv1", "in2"], tops=[], targets=[], )) xgraph.insert( XLayer( name="conv2", type=["Convolution"], bottoms=["in2"], tops=["add1"], data=ConvData( weights=np.array([[[[1, 2], [3, 4]]]], dtype=np.float32), biases=np.array([0.0, 1.0], dtype=np.float32), ), targets=[], )) xgraph.add( XLayer( name="conv3", type=["Convolution"], bottoms=["add1"], tops=[], data=ConvData( weights=np.array([[[[1, 2], [3, 4]]]], dtype=np.float32), biases=np.array([0.0, 1.0], dtype=np.float32), ), targets=[], )) xgraph.add( XLayer(name="pool1", type=["Pooling"], bottoms=["add1"], tops=[], targets=[])) xgraph.add( XLayer( name="add2", type=["Eltwise"], bottoms=["conv3", "pool1"], tops=[], targets=[], )) assert len(xgraph) == 8 assert xgraph.get_layer_names() == [ "in1", "conv1", "in2", "conv2", "add1", "conv3", "pool1", "add2", ] out_file = os.path.join(FILE_DIR, "viz.png") xgraph.visualize(out_file) os.remove(out_file)