biubug6 / Pytorch_Retinaface

Retinaface get 80.99% in widerface hard val using mobilenet0.25.
MIT License
2.63k stars 774 forks source link

想请问下 如果自己的数据集的检测框是矩形的话,应该怎么修改哈 #182

Open jackychancjcjcj opened 2 years ago

noahzhy commented 2 years ago

你是指的只检测矩形框,还是你的数据集的检测框是矩形的?如果是只需要矩形框的话,对match函数进行修改就行。如果只需要检测框是矩形的话,prior_box 部分请参考基本的ssd框架。