search

jwyang / fpn.pytorch

Pytorch implementation of Feature Pyramid Network (FPN) for Object Detection
MIT License
952 stars 221 forks source link

Specify dim in softmax function #22

Open tianlu-wang opened 6 years ago

tianlu-wang commented 6 years ago

If you are using pytorch 0.4, pls specify dim when softmax function is called in fpn.py and rpn_fpn.py, otherwise you will probably get mAP=0 for every category.

guanbin1994 commented 6 years ago

how to do that?

AshStuff commented 6 years ago

@guanbin1994 use F.softmax(tensor, 1) in both fpn.py and rpn_fpn.py

XinZhangRadar commented 5 years ago

I met this issue, it has solved follow @AshStuff advice.