alibaba/EasyCV

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

yang-ruixin opened this issue · 1 comments

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

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