【mmrotate】*** is not in the task util registry

问题:

使用mmrotate-1.x 自定义类时,明明已经注册,并添加到__init__.py中,但提示没有注册

from mmdet.registry import MODELS

@MODELS.register_module()
class RotatedATSSAssigner(BaseAssigner):

分析:

具体看提示发现,提示的是不在task util注册里,并不是常用的model模块注册里,因此需要注册到TASK_UTILS里:

from mmdet.registry import TASK_UTILS

@TASK_UTILS.register_module()
class RotatedATSSAssigner(BaseAssigner):

你可能感兴趣的:(掉过的坑,AI/ML/DL,开发语言,python)