tf/np.transpose()遇到的坑

yolov1代码中,在计算offset这一块遇到了tf.transpose()。其在一维矩阵中不起作用,在二维矩阵中起到的作用是,转置。重点来了,三维矩阵的转置,在这迟疑了好久。最后问题还是解决了。

解决方法

第一步:搞清原矩阵的shape,比如下面(2,2,4)

第二步:原矩阵顺序为(0,1,2),判定是几步转置。比如,(1,0,2),一步转置。如果是(1,2,0),就是两步转置,分成两个一步就可以了。

第三步:把转置后的位置画好。如上,还是,shape还是(2,2,4),但是,一、二维已经换了。你就把现在的相应数据的坐标,一、二维位置互换一下,互换之后找到转置前的矩阵对应坐标,就是那个数了。

>>> arr
array([[[ 0,  1,  2,  3],
        [ 4,  5,  6,  7]],

       [[ 8,  9, 10, 11],
        [12, 13, 14, 15]]])
>>> arr.transpose((1,0,2))
array([[[ 0,  1,  2,  3],
        [ 8,  9, 10, 11]],

       [[ 4,  5,  6,  7],
        [12, 13, 14, 15]]])
>>> arr.transpose((0,2,1))
array([[[ 0,  4],
        [ 1,  5],
        [ 2,  6],
        [ 3,  7]],

       [[ 8, 12],
        [ 9, 13],
        [10, 14],
        [11, 15]]])
>>>  arr.transpose((1,2,0))

  File "<pyshell#23>", line 2
    arr.transpose((1,2,0))
    ^
IndentationError: unexpected indent
>>> arr.transpose((1,2,0))
array([[[ 0,  8],
        [ 1,  9],
        [ 2, 10],
        [ 3, 11]],

       [[ 4, 12],
        [ 5, 13],
        [ 6, 14],
        [ 7, 15]]])
>>> 

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值