Huazhong University of Science and Technology

Ruofei Xiong

Qiong Liu

You Yang

The present invention discloses a scene recognition method, a training method and a device based on pyramid attention, belonging to the field of computer vision. The method includes: pyramid layering a color feature map and a depth feature map respectively, calculating the corresponding attention map of each layer; taking the output of the attention of the last layer as the output; taking the attention output of the last layer as the final feature map, for the remaining layers, adding the result after upsampling of the final feature map of an upper layer to the attention output of this layer as the final feature map of this layer; scaling the attention map and the final feature map, using the average of two new attention maps as the final attention map, mapping the largest k position in the final attention map to the final feature map of this layer.

Growing community of inventors

Hubei, China

Ruofei Xiong