Layer

Size/Stride/Padding

Output size

Input data

3 * 512 * 256

Conv + PRelu + bn

7/2/3

64 * 256 * 128

Conv + PRelu + bn

Conv + PRelu + bn

Conv + PRelu + bn

1/1/0

3/1/1

1/1/0

64 * 256 * 128

64 * 256 * 128

64 * 256 * 128

Max pooling

2/2

64 * 128 * 64

Conv + PRelu + bn

Conv + PRelu + bn

Conv + PRelu + bn

1/1/0

3/1/1

1/1/0

64 * 128 * 64

64 * 128 * 64

64 * 128 * 64

Max pooling

2/2

64 * 64 * 32

Conv + PRelu + bn

Conv + PRelu + bn

Conv + PRelu + bn

1/1/0

3/1/1

1/1/0

128 * 64 * 32

128 * 64 * 32

128 * 64 * 32