iTPN icon indicating copy to clipboard operation
iTPN copied to clipboard

fast_itpn

Open leoxxxxxD opened this issue 1 year ago • 8 comments

请问,有开源fast_itpn预训练代码的计划吗?

leoxxxxxD avatar Jan 15 '24 07:01 leoxxxxxD

可以开源。等我有空整理一下

sunsmarterjie avatar Jan 15 '24 07:01 sunsmarterjie

@sunsmarterjie 期待您的开源。还有个问题,fast_itpn目录下开源权重,是预训练权重,还是imagenet微调后的权重?

leoxxxxxD avatar Jan 15 '24 08:01 leoxxxxxD

预训练权重

---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年01月15日 16:13 | | 收件人 | @.> | | 抄送至 | @.>@.> | | 主题 | Re: [sunsmarterjie/iTPN] fast_itpn (Issue #12) |

@sunsmarterjie 期待您的开源。还有个问题,fast_itpn目录下开源权重,是预训练权重,还是imagenet微调后的权重?

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

sunsmarterjie avatar Jan 15 '24 08:01 sunsmarterjie

@sunsmarterjie 在fast_itpn里面,预训练模型转为检测模型backbone,需要做哪些修改呢(ps:检测的微调代码可以开源吗)

leoxxxxxD avatar Jan 16 '24 03:01 leoxxxxxD

替换你想用的检测器的backbone就可以了。检测代码正在整理

---- 回复的原邮件 ---- | 发件人 | @.> | | 日期 | 2024年01月16日 11:38 | | 收件人 | @.> | | 抄送至 | @.>@.> | | 主题 | Re: [sunsmarterjie/iTPN] fast_itpn (Issue #12) |

@sunsmarterjie 在fast_itpn里面,预训练模型转为检测模型backbone,需要做哪些修改呢(ps:检测的微调代码可以开源吗)

— Reply to this email directly, view it on GitHub, or unsubscribe. You are receiving this because you were mentioned.Message ID: @.***>

sunsmarterjie avatar Jan 16 '24 03:01 sunsmarterjie

@sunsmarterjie 我在下游任务使用 fast_itpn 目录下 modeling_finetune.pyFast_iTPN 作为backbone 加载预训练权重时,发现权重里有 mask_token', 'align_dim_16tofpn.weight', 'align_dim_16tofpn.bias', 'fpn_modules.0.norm2.weight', ... , 但是 Fast_iTPN 中没有定义,请问需要怎么修改呢?

pazifik-cyber avatar Mar 12 '24 02:03 pazifik-cyber

直接忽略就行。这些是预训练用到的参数,下游微调不需要加载

sunsmarterjie avatar Mar 12 '24 02:03 sunsmarterjie

好的,感谢感谢。

pazifik-cyber avatar Mar 12 '24 02:03 pazifik-cyber