mirror of
https://github.com/open-mmlab/mmocr.git
synced 2025-06-03 21:54:47 +08:00
17 lines
420 B
Python
17 lines
420 B
Python
|
# Copyright (c) OpenMMLab. All rights reserved.
|
||
|
|
from unittest import TestCase
|
||
|
|
|
||
|
|
import torch
|
||
|
|
|
||
|
|
from mmocr.models.textrecog.plugins import Maxpool2d
|
||
|
|
|
||
|
|
|
||
|
|
class TestMaxpool2d(TestCase):
|
||
|
|
|
||
|
|
def setUp(self) -> None:
|
||
|
|
self.img = torch.rand(1, 3, 32, 100)
|
||
|
|
|
||
|
|
def test_maxpool2d(self):
|
||
|
|
maxpool2d = Maxpool2d(kernel_size=2, stride=2)
|
||
|
|
self.assertEqual(maxpool2d(self.img).shape, torch.Size([1, 3, 16, 50]))
|