Skip to content

Latest commit

 

History

History
51 lines (38 loc) · 1.55 KB

README.md

File metadata and controls

51 lines (38 loc) · 1.55 KB

代码参考

对参考代码做了修改,将代码结构化,并使其可以正常训练,训练代码中没有进行测试。

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

图片与标签格式

图片名字_label_[leftup_x][leftup_y][rightdown_x]_[rightdown_y].jpg

eg: det_2018_09_13_000002_label_54_105_171_153.jpg


图片路径

训练图片均放在了dataset文件夹下面


测试图片

测试图片路径在 img 文件夹下

训练

python train.py

测试

python test.py