MCPcopy Create free account
hub / github.com/NVIDIA/DALI / setUpClass

Method setUpClass

dali/test/python/operator_1/test_debayer.py:288–307  ·  view source on GitHub ↗
(cls)

Source from the content-addressed store, hash-verified

286class DebayerVideoTest(unittest.TestCase):
287 @classmethod
288 def setUpClass(cls):
289 rng = np.random.default_rng(seed=3)
290 num_smaller, num_bigger = 4, 3
291 cls.num_samples = num_smaller + num_bigger
292 smaller = read_video(num_smaller, 60, 108, 192)
293 bigger = read_video(num_bigger, 32, 216, 384)
294 video = smaller + bigger
295 rng.shuffle(video)
296 patterns = [rng.choice(bayer_patterns, len(vid)) for vid in video]
297 cls.blue_poses = [
298 np.array([blue_position(pattern) for pattern in sample_patterns], dtype=np.int32)
299 for sample_patterns in patterns
300 ]
301 cls.bayered_vid = [
302 rgb2bayer_seq(vid, vid_patterns) for vid, vid_patterns in zip(video, patterns)
303 ]
304 cls.npp_baseline = [
305 debayer_bilinear_npp_pattern_seq(vid, vid_patterns)
306 for vid, vid_patterns in zip(cls.bayered_vid, patterns)
307 ]
308
309 @params("gpu", "cpu")
310 def test_debayer_vid_per_frame_pattern(self, device):

Callers

nothing calls this directly

Calls 4

blue_positionFunction · 0.90
rgb2bayer_seqFunction · 0.90
read_videoFunction · 0.85

Tested by

no test coverage detected