mirror of
https://github.com/open-mmlab/mmselfsup.git
synced 2025-06-03 14:59:38 +08:00
15 lines
394 B
Python
15 lines
394 B
Python
|
# Copyright (c) OpenMMLab. All rights reserved.
|
||
|
|
import pytest
|
||
|
|
import torch
|
||
|
|
|
||
|
|
from mmselfsup.utils.misc import tensor2imgs
|
||
|
|
|
||
|
|
|
||
|
|
def test_tensor2imgs():
|
||
|
|
with pytest.raises(AssertionError):
|
||
|
|
tensor2imgs(torch.rand((3, 16, 16)))
|
||
|
|
fake_tensor = torch.rand((3, 3, 16, 16))
|
||
|
|
fake_imgs = tensor2imgs(fake_tensor)
|
||
|
|
assert len(fake_imgs) == 3
|
||
|
|
assert fake_imgs[0].shape == (16, 16, 3)
|