|
|
|
|
|
|
|
|
import random |
|
|
import unittest |
|
|
|
|
|
from densepose.data.video import FirstKFramesSelector, LastKFramesSelector, RandomKFramesSelector |
|
|
|
|
|
|
|
|
class TestFrameSelector(unittest.TestCase): |
|
|
def test_frame_selector_random_k_1(self): |
|
|
_SEED = 43 |
|
|
_K = 4 |
|
|
random.seed(_SEED) |
|
|
selector = RandomKFramesSelector(_K) |
|
|
frame_tss = list(range(0, 20, 2)) |
|
|
_SELECTED_GT = [0, 8, 4, 6] |
|
|
selected = selector(frame_tss) |
|
|
self.assertEqual(_SELECTED_GT, selected) |
|
|
|
|
|
def test_frame_selector_random_k_2(self): |
|
|
_SEED = 43 |
|
|
_K = 10 |
|
|
random.seed(_SEED) |
|
|
selector = RandomKFramesSelector(_K) |
|
|
frame_tss = list(range(0, 6, 2)) |
|
|
_SELECTED_GT = [0, 2, 4] |
|
|
selected = selector(frame_tss) |
|
|
self.assertEqual(_SELECTED_GT, selected) |
|
|
|
|
|
def test_frame_selector_first_k_1(self): |
|
|
_K = 4 |
|
|
selector = FirstKFramesSelector(_K) |
|
|
frame_tss = list(range(0, 20, 2)) |
|
|
_SELECTED_GT = frame_tss[:_K] |
|
|
selected = selector(frame_tss) |
|
|
self.assertEqual(_SELECTED_GT, selected) |
|
|
|
|
|
def test_frame_selector_first_k_2(self): |
|
|
_K = 10 |
|
|
selector = FirstKFramesSelector(_K) |
|
|
frame_tss = list(range(0, 6, 2)) |
|
|
_SELECTED_GT = frame_tss[:_K] |
|
|
selected = selector(frame_tss) |
|
|
self.assertEqual(_SELECTED_GT, selected) |
|
|
|
|
|
def test_frame_selector_last_k_1(self): |
|
|
_K = 4 |
|
|
selector = LastKFramesSelector(_K) |
|
|
frame_tss = list(range(0, 20, 2)) |
|
|
_SELECTED_GT = frame_tss[-_K:] |
|
|
selected = selector(frame_tss) |
|
|
self.assertEqual(_SELECTED_GT, selected) |
|
|
|
|
|
def test_frame_selector_last_k_2(self): |
|
|
_K = 10 |
|
|
selector = LastKFramesSelector(_K) |
|
|
frame_tss = list(range(0, 6, 2)) |
|
|
_SELECTED_GT = frame_tss[-_K:] |
|
|
selected = selector(frame_tss) |
|
|
self.assertEqual(_SELECTED_GT, selected) |
|
|
|
