search

tinyvision / SOLIDER-REID

MIT License
55 stars 12 forks source link

PRETRAIN CHOICE #2

Closed LAW1223 closed 1 year ago

LAW1223 commented 1 year ago

你好，请问 MODEL.PRETRAIN_CHOICE 'self'，指的是用当前任务的数据集进行预训练，而不是imagenet吗

cwhgn commented 1 year ago

MODEL.PRETRAIN_CHOICE指的是使用什么预训练模型，如果使用solider预训练模型可以选择self，如果是使用imagenet预训练模型可以选择imagenet。

LAW1223 commented 1 year ago

您好，十分感谢您的解答。对于ReID 任务，当MODEL.PRETRAIN_CHOICE='self'，是在使用solider预训练模型。请问预训练时使用的数据集是ReID任务的数据集还是LUPerson？我之前的理解是，当MODEL.PRETRAIN_CHOICE='self'时，solider预训练模型使用ReID任务的数据集（例如MSMT17）；当MODEL.PRETRAIN_CHOICE='imagenet'时，solider预训练模型使用imagenet？

cwhgn commented 1 year ago

您好，预训练是在SOLIDER这个repo完成的，用的是LUPerson数据集，训练好的model可以在repo的readme里找到。用solider训练好的model作为预训练模型，可以在不同下游任务上使用。SOLIDER-REID这个repo是将solider训练好的预训练模型用在reid任务上finetune。Finetune时选择MODEL.PRETRAIN_CHOICE='self'即可，数据集可以选择msmt17或者market，在DATASETS.ROOT_DIR上设置。

LAW1223 commented 1 year ago

明白了，十分感谢！

获取 Outlook for iOShttps://aka.ms/o0ukef

发件人: Weihua Chen @.> 发送时间: Tuesday, April 11, 2023 7:12:28 PM 收件人: tinyvision/SOLIDER-REID @.> 抄送: Yexin LIU @.>; Author @.> 主题: Re: [tinyvision/SOLIDER-REID] PRETRAIN CHOICE (Issue #2)

你通常不会收到来自 @.*** 的电子邮件。了解这一点为什么很重要https://aka.ms/LearnAboutSenderIdentification

您好，预训练是在SOLIDERhttps://github.com/tinyvision/SOLIDER这个repo完成的，用的是LUPerson数据集，训练好的model可以在repo的readme里找到。用solider训练好的model作为预训练模型，可以在不同下游任务上使用。SOLIDER-REIDhttps://github.com/tinyvision/SOLIDER-REID这个repo是将solider训练好的预训练模型用在reid任务上finetune。Finetune时选择MODEL.PRETRAIN_CHOICE='self'即可，数据集可以选择msmt17或者market，在DATASETS.ROOT_DIR上设置。

― Reply to this email directly, view it on GitHubhttps://github.com/tinyvision/SOLIDER-REID/issues/2#issuecomment-1503138847, or unsubscribehttps://github.com/notifications/unsubscribe-auth/A25QGMHQ26SJAT5ZFDHFHDTXAU4BZANCNFSM6AAAAAAW2C5NKQ. You are receiving this because you authored the thread.Message ID: @.***>