コード例 #1
0
ファイル: bounding_box.py プロジェクト: sergeyk/skvisutils
def test_get_overlap():
    bbgt = np.array([139., 200., 69., 102.])
    bb = np.array([139., 200., 69., 102.])
    ov = BoundingBox.get_overlap(bb, bbgt)
    print(ov)
    assert(ov == 1)

    bb = np.array([139., 200., 69., 51.])
    ov = BoundingBox.get_overlap(bb, bbgt)
    print(ov)
    assert(ov == 0.5)

    bb = np.array([139., 200., 35., 51.])
    ov = BoundingBox.get_overlap(bb, bbgt)
    print(ov)
    assert((ov >= 0.24) and (ov <= 0.26))

    # switch order of arguments
    bb = np.array([139., 200., 35., 51.])
    ov = BoundingBox.get_overlap(bbgt, bb)
    print(ov)
    assert((ov >= 0.24) and (ov <= 0.26))

    bb = np.array([239., 300., 69., 51.])
    ov = BoundingBox.get_overlap(bb, bbgt)
    print(ov)
    assert(ov == 0)
コード例 #2
0
ファイル: bounding_box.py プロジェクト: sergeyk/skvisutils
def test_get_overlap_with_array():
    bbgt = np.array([139., 200., 69., 102.])
    bb1 = np.array([139., 200., 69., 102.])
    bb2 = np.array([139., 200., 69., 51.])
    bb3 = np.array([239., 300., 69., 51.])
    bb = np.vstack((bb1, bb2, bb3))
    numtimes = 100000
    bb = np.tile(bb, (numtimes, 1))
    tt = TicToc().tic()
    ov = BoundingBox.get_overlap(bb, bbgt)
    tt.toc()
    tt.tic()
    ov = BoundingBox.get_overlap(bb, bbgt)
    tt.toc()
    print(ov)
    assert(np.all(ov == np.tile(np.array([1, 0.5, 0]), numtimes)))
コード例 #3
0
ファイル: bounding_box.py プロジェクト: sergeyk/skvisutils
def test_convert_to_and_fro_with_array():
    bb1 = np.array([139., 200., 69., 102.])
    bb2 = np.array([139., 200., 69., 51.])
    bb3 = np.array([239., 300., 69., 51.])
    bb = np.vstack((bb1, bb2, bb3))
    bb = np.tile(bb, (100000, 1))
    print(bb.shape)
    tt = TicToc().tic()
    bb_c = BoundingBox.convert_arr_to_corners(bb)
    tt.toc()
    print(bb_c.shape)
    tt.tic()
    bb2 = BoundingBox.convert_arr_from_corners(bb_c)
    tt.toc()
    print(bb2.shape)
    assert(np.all(bb == bb2))
コード例 #4
0
ファイル: bounding_box.py プロジェクト: sergeyk/skvisutils
def test_convert_to_and_fro():
    bb = np.array([139., 200., 69., 102.])
    bb_c = BoundingBox.convert_arr_to_corners(bb)
    bb2 = BoundingBox.convert_arr_from_corners(bb_c)
    assert(np.all(bb == bb2))