aoru45 / fsanet.pytorch Goto Github PK

Can you explain how to run your code or add the code that you used to achieve the results that you publish?

Hi, your conversion of FSA-Net looks great, I wanted to run it in pytorch to understand the paper more.

However, when I pass an image of size: torch.Size([1, 3, 224, 224])
I get this error:

Traceback (most recent call last):
File "train_FSANet.py", line 205, in
yaw, pitch, roll = model(images)
................/lib/python3.8/site-packages/torch/nn/modules/module.py", line 550, in call
result = self.forward(input, **kwargs)
..................../FSA_net.py", line 258, in forward
feat_preS1 = self.feat_preS1(feat_s1_pre).view(-1,88)
RuntimeError: shape '[-1, 64]' is invalid for input of size 784

Please let me know if you were able to resolve this?

Kind Regards
S

你好，请问这个模型你用300W数据训练过吗？用BIWI做测试的话，精度有多少呀？