码迷,mamicode.com
首页 > 其他好文 > 详细

mask-rcnn解读(二):clip_boxes_graph()

时间:2019-11-30 18:54:34      阅读:98      评论:0      收藏:0      [点我收藏+]

标签:说明   split   turn   window   ret   函数   imu   问题   graph   

此函数是利用deltas对box修正,我并没有详细说明,若有问题,欢迎留言交流:


def clip_boxes_graph(boxes, window):
"""
boxes: [N, (y1, x1, y2, x2)]
window: [4] in the form y1, x1, y2, x2
"""
# Split
wy1, wx1, wy2, wx2 = tf.split(window, 4)
y1, x1, y2, x2 = tf.split(boxes, 4, axis=1)
# Clip
y1 = tf.maximum(tf.minimum(y1, wy2), wy1)
x1 = tf.maximum(tf.minimum(x1, wx2), wx1)
y2 = tf.maximum(tf.minimum(y2, wy2), wy1)
x2 = tf.maximum(tf.minimum(x2, wx2), wx1)
clipped = tf.concat([y1, x1, y2, x2], axis=1, name="clipped_boxes")
clipped.set_shape((clipped.shape[0], 4))
return clipped

mask-rcnn解读(二):clip_boxes_graph()

标签:说明   split   turn   window   ret   函数   imu   问题   graph   

原文地址:https://www.cnblogs.com/tangjunjun/p/11963442.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!