search

wtyhub / LPN

Pytorch implementation of Each Part Matters: Local Patterns Facilitate Cross-view Geo-localization https://arxiv.org/abs/2008.11646
MIT License
75 stars 13 forks source link

Question about the Classblock #7

Closed zkangkang0 closed 2 years ago

zkangkang0 commented 2 years ago

Hello, I've been reading this code recently. I have a problem that I don't understand.

In the model.py file, the part_classifier function defined in the three_view_net function. During the training, the structure of the Classblock in this function is as follows: train

During the test, the structure of classblock is as follows: test

I want to ask, how is this achieved?

wtyhub commented 2 years ago

@zkangkang0 This structure is achieved using these codes. https://github.com/wtyhub/LPN/blob/main/test.py#L246-L254

zkangkang0 commented 2 years ago

Thank you so much for your reply