HiBugs / CenterNet

pytorch 使用centernet进行 人脸检测Face Detection (bbox + landmark)、 船只检测SeaShips Detection

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CenterNet

使用CenterNet来训练其他数据。

本项目实现以下

  • Face 人脸bbox+landmark检测
  • SeaShips 船只种类和bbox检测

代码参考以下

Objects as Points

CenterMulti

CenterPose

结果

Face

测试:使用原图分辨率,未使用Multi Scale,在验证集上进行测试。

Backbone AP AP50 AP75 APS APM APL
DLA-34 43.7 73.7 47.7 31.9 58.8 64.0
MobileNet-V2 34.3 62.0 35.0 27.2 52.4 42.7

img

img

SeaShips

测试:使用原图分辨率,未使用Multi Scale,在验证集和测试集上进行测试。

Backbone 数据集 AP AP50 AP75 APS APM APL
DLA-34 Val 21.3 37.3 21.5 10.8 39.3 22.1
DLA-34 Test 21.6 36.4 23.5 2.7 32.4 22.8

img

训练和测试

About

pytorch 使用centernet进行 人脸检测Face Detection (bbox + landmark)、 船只检测SeaShips Detection


Languages

Language:Python 68.2%Language:Cuda 11.4%Language:C++ 10.2%Language:C 9.9%Language:Shell 0.3%Language:Makefile 0.0%