mirror of https://github.com/RE-OWOD/RE-OWOD
61 lines
1.9 KiB
Python
61 lines
1.9 KiB
Python
# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved
|
|
|
|
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)
|