Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

请问如何得到自监督MoBY算法最终predictor的输出结果,直接用于以图搜图 #260

Open
yang-ruixin opened this issue Dec 19, 2022 · 1 comment

Comments

@yang-ruixin
Copy link

Describe the feature
作者您好,当前推理时只能得到backbone的输出或者projector_q的输出,我希望可以得到最终predictor的输出,直接用于以图搜图。
在模型导出时,当前将MoBY导出为一个分类模型,舍弃了MoBY的head(即predictor),添加了一个dummy classification head。
我也可以自己实现,但是因为EasyCV包了很多层,可以实现的方式比较多,你们会官方来实现吗,或者给些建议,最elegant的实现方式。
如果我实现了可以pull request吗?
谢谢

@wuziheng
Copy link
Collaborator

  1. 开源代码用户当然可以自己选择合适的feature层来输出,我们这边只是选择了一个兼容其他自监督算法的简易处理的方案,因为这些算法有的有prejection,有的没有,所以及一般只选择了backbone.
  2. 我赞同你的pull request的建议,你可以在moby export 的时候添加额外的moby export head 的参数用于控制 moby model 导出到那一层输出。

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

No branches or pull requests

2 participants