文章讨论了在使用mmrotate-1.x时，如何正确地自定义类如RotatedATSSAssigner进行注册。作者指出，尽管在MODELS模块中进行了注册，但提示未在TASK_UTILS中找到，因此需将此类注册到TASK_UTILS模块以解决问题。
问题：
使用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):
扫一扫
被折叠的 条评论
为什么被折叠?
到【灌水乐园】发言
