[Fix] unit tests due to new config names (#1303)

This commit is contained in:
Tong Gao 2022-08-22 15:02:52 +08:00 committed by GitHub
parent 908ebf1bcf
commit 240bf06ddd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -16,7 +16,7 @@ from mmocr.utils import register_all_modules
class TestDRRG(unittest.TestCase):
def setUp(self):
cfg_path = 'textdet/drrg/drrg_r50_fpn_unet_1200e_ctw1500.py'
cfg_path = 'textdet/drrg/drrg_resnet50_fpn-unet_1200e_ctw1500.py'
self.model_cfg = self._get_detector_cfg(cfg_path)
register_all_modules()
self.model = MODELS.build(self.model_cfg)