| | import os.path as osp |
| | import unittest |
| |
|
| | import numpy as np |
| |
|
| | from mmdet.registry import TRANSFORMS |
| | from mmdet.utils import register_all_modules |
| |
|
| | register_all_modules() |
| |
|
| |
|
| | class TestInstaboost(unittest.TestCase): |
| |
|
| | def setUp(self): |
| | """Setup the model and optimizer which are used in every test method. |
| | |
| | TestCase calls functions in this order: setUp() -> testMethod() -> |
| | tearDown() -> cleanUp() |
| | """ |
| | img_path = osp.join(osp.dirname(__file__), '../../data/gray.jpg') |
| | self.results = { |
| | 'img_path': |
| | img_path, |
| | 'img_shape': (300, 400), |
| | 'instances': [{ |
| | 'bbox': [0, 0, 10, 20], |
| | 'bbox_label': 1, |
| | 'mask': [[0, 0, 0, 20, 10, 20, 10, 0]], |
| | 'ignore_flag': 0 |
| | }, { |
| | 'bbox': [10, 10, 110, 120], |
| | 'bbox_label': 2, |
| | 'mask': [[10, 10, 110, 10, 110, 120, 110, 10]], |
| | 'ignore_flag': 0 |
| | }, { |
| | 'bbox': [50, 50, 60, 80], |
| | 'bbox_label': 2, |
| | 'mask': [[50, 50, 60, 50, 60, 80, 50, 80]], |
| | 'ignore_flag': 1 |
| | }] |
| | } |
| |
|
| | def test_transform(self): |
| | load = TRANSFORMS.build(dict(type='LoadImageFromFile')) |
| | instaboost_transform = TRANSFORMS.build(dict(type='InstaBoost')) |
| |
|
| | |
| | results = load(self.results) |
| | results = instaboost_transform(results) |
| |
|
| | self.assertEqual(results['img'].dtype, np.uint8) |
| | self.assertIn('instances', results) |
| |
|
| | def test_repr(self): |
| | instaboost_transform = TRANSFORMS.build(dict(type='InstaBoost')) |
| |
|
| | self.assertEqual( |
| | repr(instaboost_transform), 'InstaBoost(aug_ratio=0.5)') |
| |
|
