如何加载预训练模型？

zcablii / SARDet_100K

Offical implementation of MSFA and release of SARDet_100K dataset for Large-Scale Synthetic Aperture Radar (SAR) Object Detection

Other

328 stars 25 forks source link

Closed EEJQX closed 4 months ago

EEJQX commented 5 months ago

如果我想使用您训练好的模型，从哪里寻找对应模型的代码呢？我貌似没看到模型权值文件对应的实例化代码。

zcablii commented 5 months ago

通过配置文件修改中的加载模型权重文件的路径。默认的配置文件放在了local_configs下。更多关于配置文件的说明可参考mmdet

EEJQX commented 5 months ago

感谢回复！

EEJQX commented 5 months ago

KeyError: 'Self_features_model is not in the mmdet::model registry.

EEJQX commented 5 months ago

没用过mmdet库，感觉不太好上手哎

zcablii commented 5 months ago

KeyError: 'Self_features_model is not in the mmdet::model registry.

你好像没用最新版本的repo。之前版本的代码里历史遗留问题导致Self_features_model的报错，最新版本的代码已经修复这个问题了。