Hi how to train your code with different pre-trained models about person-reid-triplet-loss-baseline HOT 4 OPEN

On Wed, 3 Mar 2021, 1:34 am Houjing Huang, ***@***.***> wrote: Hi, to use another backbone. - First, you create another file, e.g. resnext.py under directory tri_loss/model. The content is the original resnext.py provided by pytorch. - Then, modify the forward function of resnext, removing unnecessary operation after this line x = self.layer4(x), so that it returns the result of layer4. - Finally, you can use from .resnext import resnext50 in tri_loss/model/Model.py, and then replace this line self.base = resnet50(pretrained=True, last_conv_stride=last_conv_stride) with self.base = resnext50(pretrained=True) If you would like to reduce the last convolutional stride of the backbone, you can modify it yourself in resnext.py accordingly. — You are receiving this because you authored the thread. Reply to this email directly, view it on GitHub <#43 (comment)>, or unsubscribe <https://github.com/notifications/unsubscribe-auth/AOOYJBPSEOJLQ5H5234LWQLTBVAEZANCNFSM4YOF3KGQ> .

Thank you very much for your clear explanation. I want to check with different networks like Efficient net, Inception. Where do I get pretrained weight for these networks. Can you suggest which pretrained network is preferred for reidentification. Thank you. On Wed, 3 Mar 2021, 1:34 am Houjing Huang, ***@***.***> wrote: > Hi, to use another backbone. > > - First, you create another file, e.g. resnext.py under directory > tri_loss/model. The content is the original resnext.py provided by > pytorch. > - Then, modify the forward function of resnext, removing unnecessary > operation after this line x = self.layer4(x), so that it returns the > result of layer4. > - Finally, you can use from .resnext import resnext50 in > tri_loss/model/Model.py, and then replace this line > > self.base = resnet50(pretrained=True, last_conv_stride=last_conv_stride) > > with > > self.base = resnext50(pretrained=True) > > If you would like to reduce the last convolutional stride of the > backbone, you can modify it yourself in resnext.py accordingly. > > — > You are receiving this because you authored the thread. > Reply to this email directly, view it on GitHub > <#43 (comment)>, > or unsubscribe > <https://github.com/notifications/unsubscribe-auth/AOOYJBPSEOJLQ5H5234LWQLTBVAEZANCNFSM4YOF3KGQ> > . >