mirror of https://github.com/open-mmlab/mmocr.git
parent
2ad699e099
commit
b153cc01c8
|
@ -261,7 +261,6 @@ def test_psenet(cfg_file):
|
|||
detector.show_result(img, results)
|
||||
|
||||
|
||||
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
|
||||
@pytest.mark.parametrize('cfg_file', [
|
||||
'textdet/dbnet/dbnet_r18_fpnc_1200e_icdar2015.py',
|
||||
'textdet/dbnet/dbnet_r50dcnv2_fpnc_1200e_icdar2015.py'
|
||||
|
|
|
@ -18,7 +18,6 @@ def test_char_attn():
|
|||
assert out_feat_map.shape == torch.Size([1, 128, 32, 32])
|
||||
|
||||
|
||||
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
|
||||
def test_feat_generator():
|
||||
in_feat = torch.rand(1, 128, 32, 32)
|
||||
feat_generator = FeatGenerator(in_channels=128, out_channels=128)
|
||||
|
@ -28,7 +27,6 @@ def test_feat_generator():
|
|||
assert feat_map.shape == torch.Size([1, 128, 32, 32])
|
||||
|
||||
|
||||
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
|
||||
def test_cafcn_neck():
|
||||
in_s1 = torch.rand(1, 64, 64, 64)
|
||||
in_s2 = torch.rand(1, 128, 32, 32)
|
||||
|
|
|
@ -81,7 +81,6 @@ def test_base_recognizer():
|
|||
tmp_dir.cleanup()
|
||||
|
||||
|
||||
@pytest.mark.skip(reason='TODO: re-enable after CI support pytorch>1.4')
|
||||
def test_seg_recognizer():
|
||||
tmp_dir = tempfile.TemporaryDirectory()
|
||||
# create dummy data
|
||||
|
|
Loading…
Reference in New Issue