search

polarisZhao / PFLD-pytorch

PFLD pytorch Implementation
798 stars 197 forks source link

Can someone help me to understand "self.fc = nn.Linear(176, 196)" how 176 input feature has been choose #75

Open sainisanjay opened 2 years ago

sainisanjay commented 2 years ago

Can someone help me to understand self.fc = nn.Linear(176, 196) how 176 input feature has been choose in PFLDInference network.

Sridhar98 commented 1 year ago

I have the same question. Do you have any idea about this @sainisanjay ?

CrossEntropy commented 1 year ago

That's because when you stack three multi-scale features together, you get a new tensor which shape is (bs, 176). Please see the code in models/pfld.py @sainisanjay @Sridhar98