search

jwyang / fpn.pytorch

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

torch.log(torch.sqrt(h * w) / 224.0) in fpn.py #49

Open fafancier opened 4 years ago

fafancier commented 4 years ago

Hi, Thanks for sharing. I found a line in fpn.py. "roi_level = torch.log(torch.sqrt(h * w) / 224.0)" https://github.com/jwyang/fpn.pytorch/blob/23bd1d2fa09fbb9453f11625d758a61b9d600942/lib/model/fpn/fpn.py#L110 It seems torch.log is based on the e rather than 2 in the paper. Is that true?

CharlesPikachu commented 4 years ago

I think you could set torch.log2 instead