代码参考

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

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