当前位置:网站首页>PSP代码实现

PSP代码实现

2022-06-23 06:18:00 休斯顿凤梨

原理:

      Backbone+PSP+Prediction

实现:

#代码参考官方
import paddle.fluid as fluid
import numpy as np 
np.set_printoptions(precision=2)

x = [ [0.14,0.8,0.34,0.2,0.004,0.31,0.34,0.11],
      [0.69,0.51,0.35,0.11,0.22,0.47,0.39,0.92],
      [0.64,0.77,0.41,0.08,0.86,0.61,0.41,0.6],
      [0.35,0.9,0.07,0.92,0.16,0.78,0.17,0.87],
      [0.95,0.44,0.15,0.53,0.89,0.38,0.51,0.08],
      [0.93,0.64,0.83,0.37,0.56,0.37,0.68,0.66],
      [0.37,0.01,0.58,0.2,0.81,0.75,0.05,0.15],
      [0.08,0.66,0.63,0.25,0.18,0.67,0.97,0.34]]

def main():
    global x
    with fluid.dygraph.guard(fluid.CPUPlace()):
        x = np.array(x)
        print(x.shape)
        print(x)
        x = x[np.newaxis,np.newaxis,:,:]
        x = fluid.dygraph.to_variable(x)
        y = fluid.layers.adaptive_pool2d(input=x,pool_size=[5,3],pool_type='max')
     
        y = y.numpy().squeeze((0,1))
        print(y.shape)
        print(y)
        
if __name__ == "__main__":
    main()




    




效果

 

 

原网站

版权声明
本文为[休斯顿凤梨]所创,转载请带上原文链接,感谢
https://blog.csdn.net/weixin_45779396/article/details/124942118