コード例 #1
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)))
コード例 #2
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))