utils.py 文件源码

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)