def test_bbox_transform():
gt_rois = numpy.array([[-84., -40., 99., 55.], [-176., -88., 191., 103.],
[-360., -184., 375., 199.], [-56., -56., 71., 71.],
[-120., -120., 135., 135.],
[-248., -248., 263., 263.], [-36., -80., 51., 95.],
[-80., -168., 95., 183.],
[-168., -344., 183., 359.]])
ex_rois = 2 * gt_rois
gt_rois = keras.backend.variable(gt_rois)
ex_rois = keras.backend.variable(ex_rois)
results = keras_rcnn.backend.bbox_transform(ex_rois, gt_rois)
results = keras.backend.eval(results)
expected = numpy.array(
[[-0.02043597, -0.03926702, -0.69042609, -0.68792524],
[-0.01020408, -0.01958225, -0.69178756, -0.69053962],
[-0.00509857, -0.00977836, -0.6924676, -0.69184425],
[-0.02941176, -0.02941176, -0.68923328, -0.68923328],
[-0.0146771, -0.0146771, -0.69119215, -0.69119215],
[-0.00733138, -0.00733138, -0.69217014, -0.69217014],
[-0.04285714, -0.02136752, -0.68744916, -0.69030223],
[-0.02136752, -0.01066856, -0.69030223, -0.69172572],
[-0.01066856, -0.00533049, -0.69172572, -0.6924367]])
numpy.testing.assert_array_almost_equal(results, expected)
评论列表
文章目录
