test_image_classification_layer.py 2.9 KB
Newer Older
D
dzhwinter 已提交
1
#   Copyright (c) 2018 PaddlePaddle Authors. All Rights Reserve.
D
dzhwinter 已提交
2
#
D
dzhwinter 已提交
3 4 5
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
D
dzhwinter 已提交
6
#
D
dzhwinter 已提交
7
#     http://www.apache.org/licenses/LICENSE-2.0
D
dzhwinter 已提交
8
#
D
dzhwinter 已提交
9 10 11 12 13 14
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.

15 16
import unittest

17
import paddle.v2.fluid as fluid
18 19
import paddle.v2.fluid.nets as nets
from paddle.v2.fluid.framework import Program
20 21


22
def conv_block(input, num_filter, groups, dropouts):
23 24 25 26 27 28 29 30 31
    return nets.img_conv_group(
        input=input,
        pool_size=2,
        pool_stride=2,
        conv_num_filter=[num_filter] * groups,
        conv_filter_size=3,
        conv_act='relu',
        conv_with_batchnorm=True,
        conv_batchnorm_drop_rate=dropouts,
32
        pool_type='max')
33 34 35 36


class TestLayer(unittest.TestCase):
    def test_batch_norm_layer(self):
37 38
        main_program = Program()
        startup_program = Program()
39 40 41 42 43 44
        with fluid.program_guard(main_program, startup_program):
            images = fluid.layers.data(
                name='pixel', shape=[3, 48, 48], dtype='float32')
            hidden1 = fluid.layers.batch_norm(input=images)
            hidden2 = fluid.layers.fc(input=hidden1, size=128, act='relu')
            fluid.layers.batch_norm(input=hidden2)
45

46
        print str(main_program)
47 48

    def test_dropout_layer(self):
49 50
        main_program = Program()
        startup_program = Program()
51 52 53 54
        with fluid.program_guard(main_program, startup_program):
            images = fluid.layers.data(
                name='pixel', shape=[3, 48, 48], dtype='float32')
            fluid.layers.dropout(x=images, dropout_prob=0.5)
55

56
        print str(main_program)
57 58

    def test_img_conv_group(self):
59 60
        main_program = Program()
        startup_program = Program()
61

62 63 64 65 66
        with fluid.program_guard(main_program, startup_program):
            images = fluid.layers.data(
                name='pixel', shape=[3, 48, 48], dtype='float32')
            conv1 = conv_block(images, 64, 2, [0.3, 0])
            conv_block(conv1, 256, 3, [0.4, 0.4, 0])
67

68
        print str(main_program)
69

Q
Qiao Longfei 已提交
70
    def test_elementwise_add_with_act(self):
71 72
        main_program = Program()
        startup_program = Program()
73 74 75 76 77 78 79
        with fluid.program_guard(main_program, startup_program):
            image1 = fluid.layers.data(
                name='pixel1', shape=[3, 48, 48], dtype='float32')
            image2 = fluid.layers.data(
                name='pixel2', shape=[3, 48, 48], dtype='float32')
            fluid.layers.elementwise_add(x=image1, y=image2, act='relu')
        print(main_program)
Q
Qiao Longfei 已提交
80

81 82 83

if __name__ == '__main__':
    unittest.main()