def test_vgg(self): img = layer.data(name='pixel', type=data_type.dense_vector(784)) vgg_out = networks.small_vgg(input_image=img, num_channels=1, num_classes=2) print layer.parse_network(vgg_out)
def test_vgg(self): img = layer.data(name='pixel1', type=data_type.dense_vector(784)) vgg_out = networks.small_vgg( input_image=img, num_channels=1, num_classes=2) print layer.parse_network(vgg_out)