/DenseBox

Implemention of Baidu's DenseBox used for multi-task learning of object detection and landmark(key-point) localization 用PyTorch实现了百度的DenseBox并针对任意宽高比矩形(不仅限于方形)的目标检测做了优化,不仅可以输出关键点的热力图(heatmap)而且可以输出每个bbox对应关键点坐标

Primary LanguagePythonMIT LicenseMIT

DenseBox

Baidu's Densebox implemention with PyTorch used for multi-task learning of object detection and landmark(key-point) localization

Test result




Perspective transformation result




A small dataset sample:

patches

extract code: ir6n