search

louis-she / sfd.pytorch

S3FD: single shot face detector in pytorch
MIT License
114 stars 20 forks source link

Change Anchor Boxes Aspect Ratio #3

Open ahkarami opened 6 years ago

ahkarami commented 6 years ago

Dear @louis-she, Thank you for your nice work. As Shifeng Zhang has mentioned in his paper: Our anchors are 1:1 aspect ratio (_i.e.,_ square anchor), the aspect ratios of all anchor boxes are 1:1. However, if one wants to change the aspect ratios of anchor boxes for example, to 1:1.4 (width:height) the network structure must be changed or just the training code & decoding strategy (i.e., utils.py or anchor.py) must be changed? Would you please explain how one can change the aspect ratios of anchor boxes?

louis-she commented 6 years ago

Currently the aspect ratio of anchors is hard code to 1:1. But I already have plans to support other object detection(not only for "square" faces) so the aspect ratio of anchors will be necessary. Since then you will be able to change the aspect ratio of anchors. Thanks.

louis-she commented 6 years ago

This issue should be opened until the ANCHOR_RATIO been added to config.py

ahkarami commented 6 years ago

@louis-she, Thank you for your time & response.

ahkarami commented 6 years ago

Dear @louis-she, Do you have any plan for adding ANCHOR_RATIO in near future?

louis-she commented 6 years ago

I'm now working on improving the behavior of the model ( mainly about the mAP ). If you are interested in this, a PR is welcomed.