search

open-mmlab / mmrotate

OpenMMLab Rotated Object Detection Toolbox and Benchmark
https://mmrotate.readthedocs.io/en/latest/
Apache License 2.0
1.88k stars 558 forks source link

[New Models] 自己实现的检测头权重在评估过程中无法正确加载 #1059

Open UncleNiNi opened 3 months ago

UncleNiNi commented 3 months ago

Model/Dataset/Scheduler description

我自己实现了一个自定义的bbox头，在这个实现bbox头的文件里我自定义了一个模块。 使用了我的自定义bbox头的模型训练过程是正常的，但是在评估过程中，我发现模型表现不正常，经过排查之后发现是因为我在bbox头文件里自定义模块的训练好的权重在评估中没有被加载，而是随机初始化的。请问我该如何解决这个问题？

Open source status

Provide useful links for the implementation

No response