def flip_gt_boxes(gt_boxes, ih, iw):
x1s, y1s, x2s, y2s, cls = \
gt_boxes[:, 0], gt_boxes[:, 1], gt_boxes[:, 2], gt_boxes[:, 3], gt_boxes[:, 4]
x1s = tf.to_float(iw) - x1s
x2s = tf.to_float(iw) - x2s
return tf.concat(values=(x2s[:, tf.newaxis],
y1s[:, tf.newaxis],
x1s[:, tf.newaxis],
y2s[:, tf.newaxis],
cls[:, tf.newaxis]), axis=1)
评论列表
文章目录