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)
