|
|
|
from unittest import TestCase |
|
|
|
import numpy as np |
|
|
|
from mmpose.datasets.datasets.base import (Kpt2dSviewRgbImgTopDownDataset, |
|
Kpt2dSviewRgbVidTopDownDataset, |
|
Kpt3dSviewRgbImgTopDownDataset) |
|
|
|
|
|
class TestDatasetCompatibility(TestCase): |
|
|
|
def test_xywh2cs(self): |
|
|
|
base_dataset_list = [ |
|
Kpt2dSviewRgbImgTopDownDataset, Kpt2dSviewRgbVidTopDownDataset, |
|
Kpt3dSviewRgbImgTopDownDataset |
|
] |
|
for basetype in base_dataset_list: |
|
|
|
class DummyDataset(basetype): |
|
|
|
def __init__(self): |
|
self.ann_info = dict(image_size=[192, 256]) |
|
self.test_mode = True |
|
|
|
def _get_db(self): |
|
pass |
|
|
|
def evaluate(self, results, *args, **kwargs): |
|
pass |
|
|
|
dataset = DummyDataset() |
|
|
|
with self.assertWarnsRegex( |
|
DeprecationWarning, |
|
'The ``_xywh2cs`` method will be deprecated'): |
|
bbox = np.array([0, 0, 100, 100], dtype=np.float32) |
|
_ = dataset._xywh2cs(*bbox) |
|
|