Robert-JunWang / Pelee

Pelee: A Real-Time Object Detection System on Mobile Devices
Apache License 2.0
885 stars 254 forks source link

peele-SSD add_extra_layers_pelee #69

Closed qianlinjun closed 5 years ago

qianlinjun commented 5 years ago

hello I saw your comment in the function of add_extra_layers_pelee in feature_extractor.py: `

# stage2_tb: 38 x 38 x 256
# stage3_tb: 19 x 19 x 512
# stage4_tb: 10 x 10 x 704 `

however, assume input_image size is [304, 304], I found output size are as folows: `

stem_block: 76 x 76 x 32

# stage1_tb: 38 x 38 x 128
# stage2_tb: 19 x 19 x 256
# stage3_tb: 10 x 10 x 512
# stage4_tb: 10 x 10 x 704 `

is the comment wrong?