问题:
使用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):