deepblue0822 / spp_net

A SPP_net in pytorch framework

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

pytorch_prac

A SPP_net in pytorch framework which could be add to CNN arch freely.

SPP layer could be added in CNN model between convolutional layer and fully-connected lay, so that you can input multi-size images into your CNN model.

The class SPPLayer in file SPP_layer.py is independent. It could be added in your own models.

paper

Kaiming He, Xiangyu Zhang, Shaoqing Ren, Jian Sun 《Spatial Pyramid Pooling in Deep Convolutional Networks for Visual Recognition》

blog

空间金字塔池化(Spatial Pyramid Pooling, SPP)原理和代码实现(Pytorch)

About

A SPP_net in pytorch framework


Languages

Language:Python 100.0%